diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index e164a74d21034fc723752d59d4d56e004e9bee26..a53b16699c9e0da49720df0f638cde47e8b65119 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -14,7 +14,7 @@ logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging
 NUM_WU_EPOCHS = 1
 NUM_EPOCHS = 20
 num_chans = 2
-PSNR_MAX = 1.0
+PSNR_MAX = 256.0
 
 class Trainer(object):
   """ Trainer class for ESRGAN """