From bb0d9fe7780467edddc04e7236f6480410437058 Mon Sep 17 00:00:00 2001 From: Amirsina Torfi Date: Mon, 17 Jul 2017 10:34:44 -0400 Subject: [PATCH] Update train.py --- code/training_evaluation/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/training_evaluation/train.py b/code/training_evaluation/train.py index d2c39e9..62737a6 100644 --- 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.')