diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 55581937d25115ad7bef67a06896148792d1d262..14518f3b3743dfa02996d78cc1f9af2914c1ad1f 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -13,7 +13,7 @@ import h5py LOG_DEVICE_PLACEMENT = False -PROC_BATCH_SIZE = 4 +PROC_BATCH_SIZE = 2 PROC_BATCH_BUFFER_SIZE = 50000 NumClasses = 2