diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 905e39589902ebabced5ed18901f54c238b4765b..aea68231617991bf85502043a85b784718045bc5 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 = 60
-NUM_EPOCHS = 80
+NUM_WU_EPOCHS = 1
+NUM_EPOCHS = 20
 num_chans = 2