diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index be6d471e48a026a8b5e9c069587b8c7aef43df85..2b972f45537c6c45f4a2ecd57db1991c05f1f1f5 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -13,7 +13,7 @@ import h5py
 
 LOG_DEVICE_PLACEMENT = False
 
-PROC_BATCH_SIZE = 10
+PROC_BATCH_SIZE = 4
 PROC_BATCH_BUFFER_SIZE = 50000
 
 NumClasses = 2
@@ -801,7 +801,7 @@ class ESPCN:
         train_data_files.sort()
         valid_data_files.sort()
 
-        self.setup_pipeline(train_data_files, valid_data_files, 200000)
+        self.setup_pipeline(train_data_files, valid_data_files, 50000)
         self.build_model()
         self.build_training()
         self.build_evaluation()