Update main.py

main.py

@@ -521,7 +521,7 @@ def main_func(target, experiment, cfg_arg):
 
             ### added for app
             if int(t)%30==0:
-                print(f"debug: {t}, {filename} {img.size()}")
+                # print(f"debug: {t}, {filename} {img.size()}")
                 return img.detach().cpu().numpy(), int(t)
 
             x = img.unsqueeze(0).permute(0, 3, 1, 2) # HWC -> NCHW