diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 5b21942b31c28176b405eef20dcf14821808dced..2cced8eb3d9ca9d6556ca00e7765b75b53aec9b5 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -11,7 +11,7 @@ from pathlib import Path
 
 logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging.DEBUG)
 
-NUM_WU_EPOCHS = 10
+NUM_WU_EPOCHS = 8
 NUM_EPOCHS = 20
 num_chans = 2
 PSNR_MAX = 256.0