diff --git a/src/cryo_sbi/inference/losses.py b/src/cryo_sbi/inference/losses.py index f31ea09..7ed6391 100644 --- a/src/cryo_sbi/inference/losses.py +++ b/src/cryo_sbi/inference/losses.py @@ -50,6 +50,5 @@ def forward(self, theta: torch.Tensor, x: torch.Tensor, x_obs: torch.Tensor) -> self.estimator.train() log_p = self.estimator.npe(self.estimator.standardize(theta), latent_vecs_x) - print(-log_p.mean().item(), summary_stats_regularization.mean().item()) return -log_p.mean() + 0.0 * summary_stats_regularization \ No newline at end of file