diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 4bcbdc5e71fff60e7bdeb89be9c8df3d2cd6d6d1..2973c69ce9f00149f41540f8ece013d5a4eeab0c 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -13,7 +13,7 @@ import h5py LOG_DEVICE_PLACEMENT = False -PROC_BATCH_SIZE = 2 +PROC_BATCH_SIZE = 4 PROC_BATCH_BUFFER_SIZE = 50000 NumClasses = 2