removed print command

flatironinstitute · Jun 24, 2024 · b098816 · b098816
1 parent d819cc8
commit b098816
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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