snapshot...

modules/GSOC/E2_ESRGAN/lib/train.py

@@ -106,7 +106,6 @@ class Trainer(object):
       return distributed_metric
 
     for image_lr, image_hr in self.dataset:
-      print(image_lr.shape, image_hr.shape)
       num_steps = train_step(image_lr, image_hr)
 
       if num_steps >= total_steps:
@@ -262,7 +261,6 @@ class Trainer(object):
     last_psnr = 0
 
     for image_lr, image_hr in self.dataset:
-      print(image_lr.shape, image_hr.shape)
       num_step = train_step(image_lr, image_hr)
       if num_step >= total_steps:
         return