diff --git a/modules/GSOC/E2_ESRGAN/main.py b/modules/GSOC/E2_ESRGAN/main.py
index d78209ec8ceeac1e702773fb6a84295674d780eb..26abe90bbe07403fbb69680ae37778a4971a5a1e 100644
--- a/modules/GSOC/E2_ESRGAN/main.py
+++ b/modules/GSOC/E2_ESRGAN/main.py
@@ -102,7 +102,7 @@ def run(kwargs):
         partial(model.RRDBNet, out_channel=1, first_call=False),
         discriminator,
         sett["interpolation_parameter"],
-        [720, 1080],
+        [512, 512],
         basepath=kwargs["model_dir"])
 
     tf.saved_model.save(