[mnist]: Address comments in PR

asimshankar · asimshankar · commit 6daaec16e983 · 2018-01-10T12:25:14.000-08:00
diff --git a/official/mnist/mnist_tpu.py b/official/mnist/mnist_tpu.py
@@ -30,7 +30,7 @@
 tf.flags.DEFINE_string("data_dir", "",
                        "Path to directory containing the MNIST dataset")
 tf.flags.DEFINE_string("model_dir", None, "Estimator model_dir")
-tf.flags.DEFINE_integer("batch_size", 128,
+tf.flags.DEFINE_integer("batch_size", 1024,
                         "Mini-batch size for the training. Note that this "
                         "is the global batch size and not the per-shard batch.")
 tf.flags.DEFINE_integer("train_steps", 1000, "Total number of training steps.")
