diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index a53b16699c9e0da49720df0f638cde47e8b65119..5b21942b31c28176b405eef20dcf14821808dced 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -11,11 +11,12 @@ from pathlib import Path
 
 logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging.DEBUG)
 
-NUM_WU_EPOCHS = 1
+NUM_WU_EPOCHS = 10
 NUM_EPOCHS = 20
 num_chans = 2
 PSNR_MAX = 256.0
 
+
 class Trainer(object):
   """ Trainer class for ESRGAN """