diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 5f63ef37d62644eee2f6db496acb83bd4d15d5f7..a795846db468afc17232c58826258d7d61737db5 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -22,7 +22,7 @@ if NumClasses == 2: else: NumLogits = NumClasses -BATCH_SIZE = 64 +BATCH_SIZE = 128 NUM_EPOCHS = 80 TRACK_MOVING_AVERAGE = False