snapshot...

modules/GSOC/E2_ESRGAN/lib/train.py

@@ -109,6 +109,7 @@ class Trainer(object):
       return distributed_metric
 
     for epoch in range(NUM_WU_EPOCHS):
+        print('start epoch #: ', epoch)
         for image_lr, image_hr in self.dataset:
           num_steps = train_step(image_lr, image_hr)
 
@@ -263,6 +264,7 @@ class Trainer(object):
     last_psnr = 0
 
     for epoch in range(NUM_EPOCHS):
+        print('start epoch #: ', epoch)
         for image_lr, image_hr in self.dataset:
           num_step = train_step(image_lr, image_hr)