diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py index 22fef508582a5c0822c13be958c163f556a1d2d6..ac48738a53299623fb33eed1c55cc6ddc81aba61 100644 --- a/modules/GSOC/E2_ESRGAN/lib/train.py +++ b/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