diff --git a/modules/GSOC/E2_ESRGAN/config/config.yaml b/modules/GSOC/E2_ESRGAN/config/config.yaml
index d85944b517c1f4eb3536a39af2991508870cbba2..c70e810f4365e771a31362053efbd556fc1d6bf7 100644
--- a/modules/GSOC/E2_ESRGAN/config/config.yaml
+++ b/modules/GSOC/E2_ESRGAN/config/config.yaml
@@ -16,7 +16,7 @@ RDB:
   residual_scale_beta: 0.2
 #Training
 # warmup_num_iter: 1000 # Number of steps to train the PSNR model
-batch_size: 32
+batch_size: 4
 train_psnr:
   num_steps: 600000
   adam: