diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index eae290fb36cf4924140d1176978e22ec053726e5..60fdb546b9fddea93868acf9317fb0e7a2e37405 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -22,7 +22,7 @@ if NumClasses == 2: else: NumLogits = NumClasses -BATCH_SIZE = 64 +BATCH_SIZE = 128 NUM_EPOCHS = 80 TRACK_MOVING_AVERAGE = False