diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index a0498295fb229bb66a5288e4bc8522e1b5c49a05..9e13045034400f042d08adee90d3ffa535aa5622 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/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)