diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 2cced8eb3d9ca9d6556ca00e7765b75b53aec9b5..7d234fcd03c2672bfff325c857eae48b086a22d6 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 = 8
+NUM_WU_EPOCHS = 14
 NUM_EPOCHS = 20
 num_chans = 2
 PSNR_MAX = 256.0