snapshot...

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()