perform grey box parameter training using only initial likelihood.

lasgroup · Nov 9, 2023 · 795896f · 795896f
1 parent da14141
commit 795896f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sim_transfer/models/bnn_fsvd_grey_box_model.py b/sim_transfer/models/bnn_fsvd_grey_box_model.py
@@ -117,8 +117,7 @@ def _sim_loss(self, params_sim: NamedTuple, params_nn: Dict, x_batch: jnp.array,
                                                               normalized_y=True)
 
         # get likelihood std
-        likelihood_std = self._likelihood_std_transform(params_nn['likelihood_std_raw']) if self.learn_likelihood_std \
-            else self.likelihood_std
+        likelihood_std = self.likelihood_std
 
         def _ll(pred, y):
             return tfd.MultivariateNormalDiag(pred, likelihood_std).log_prob(y)