Skip to content

GitLab

Commit 06dd5c7e authored by Asim Shankar's avatar Asim Shankar
Browse files

Respond to PR comments

parent fd470cae
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/mnist/mnist_eager.py
View file @ 06dd5c7e
......@@ -104,7 +104,7 @@ def main(_):
# Load the datasets
train_ds = dataset.train(FLAGS.data_dir).shuffle(60000).batch(
FLAGS.batch_size)
test_ds = dataset.test(FLAGS.data_dir).batch(10000)
test_ds = dataset.test(FLAGS.data_dir).batch(FLAGS.batch_size)
# Create the model and optimizer
model = mnist.Model(data_format)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment