diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index 1f1d9c8a9945b75d8e36e0b3b3a7ae7302b2f4f4..5207706779baf267ec873124e54e47261828dd5b 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 = 30
 NUM_EPOCHS = 20