diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index be27c00076206c01ea3f1ef7eb6daae71038bde6..d14789811ff8ba04fc7e79ac9a530bc1d5ad27f2 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -18,7 +18,7 @@ PROC_BATCH_SIZE = 4096 PROC_BATCH_BUFFER_SIZE = 50000 NumClasses = 2 NumLogits = 1 -BATCH_SIZE = 256 +BATCH_SIZE = 128 NUM_EPOCHS = 100 TRACK_MOVING_AVERAGE = False