FIX: image debug log disable

GunwooHan · Jun 6, 2024 · 79132bb · 79132bb
1 parent 6c28234
commit 79132bb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/data/latent_twoafc.py b/data/latent_twoafc.py
@@ -37,9 +37,9 @@ def __getitem__(self, idx):
         judge_img = torch.FloatTensor(judge_img)
 
         # For Debug
-        import torchvision.utils as vutils
-        temp_tensor = [p0_img, p1_img, ref_img]
-        vutils.save_image(temp_tensor, "output.png", nrow=3, normalize=True)
+        # import torchvision.utils as vutils
+        # temp_tensor = [p0_img, p1_img, ref_img]
+        # vutils.save_image(temp_tensor, "output.png", nrow=3, normalize=True)
 
         return p0_img, p1_img, ref_img, judge_img