diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 4294d634fcbd3cc04fc93ad943243fe31f18f5ca..a6775e1c73bc95b62f84c1e8909d77ba82458f30 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -11,8 +11,8 @@ from pathlib import Path
 
 logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging.DEBUG)
 
-NUM_WU_EPOCHS = 30
-NUM_EPOCHS = 30
+NUM_WU_EPOCHS = 60
+NUM_EPOCHS = 80
 
 
 class Trainer(object):