diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index b1278ab1666ed692faa844af914880db158fd96c..fc992c7c463248266309aa984683479af7b93d72 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -63,6 +63,7 @@ class Trainer(object):
     decay_step = decay_params["step"]
     decay_factor = decay_params["factor"]
     total_steps = phase_args["num_steps"]
+    print('total number of steps: ', total_steps)
     metric = tf.keras.metrics.Mean()
     psnr_metric = tf.keras.metrics.Mean()