From b098816b5a5986db56f48a365c94e0fe139cc3e0 Mon Sep 17 00:00:00 2001 From: Dingel321 <Dingel321@gmail.com> Date: Mon, 24 Jun 2024 16:37:15 -0400 Subject: [PATCH] removed print command --- src/cryo_sbi/inference/losses.py | 1 - 1 file changed, 1 deletion(-) 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