Remove print statements from lensing_encoder

prob-ml · Aug 19, 2024 · 93b1beb · 93b1beb
1 parent 134aaf8
commit 93b1beb
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/case_studies/weak_lensing/lensing_encoder.py b/case_studies/weak_lensing/lensing_encoder.py
@@ -77,10 +77,6 @@ def _compute_loss(self, batch, logging_name):
         batch_size, _, _, _ = batch["images"].shape[0:4]
 
         target_cat = BaseTileCatalog(batch["tile_catalog"])
-        print("shear", target_cat["shear"].shape)
-        print("convergence", target_cat["convergence"].shape)
-        print("ellipticity", target_cat["ellip_lensed"].shape)
-        print("shear nans: ", torch.isnan(target_cat["shear"]).any())
 
         # multiple image normalizers
         input_lst = [inorm.get_input_tensor(batch) for inorm in self.image_normalizers]