Update train.py

astorfi · Jul 17, 2017 · bb0d9fe · bb0d9fe
1 parent d433948
commit bb0d9fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/training_evaluation/train.py b/code/training_evaluation/train.py
@@ -111,7 +111,7 @@
 
 
 tf.app.flags.DEFINE_integer(
-    'batch_size', 128, 'The number of samples in each batch.')
+    'batch_size', 32, 'The number of samples in each batch.')
 
 tf.app.flags.DEFINE_integer(
     'num_epochs', 1, 'The number of epochs for training.')