Skip to content

GitLab

Commit 2b56ede5 authored by Asim Shankar's avatar Asim Shankar
Browse files

Missed comment.

parent ddf382bb
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/mnist/mnist_eager.py
View file @ 2b56ede5
...@@ -129,6 +129,7 @@ def main(_): ...@@ -129,6 +129,7 @@ def main(_):
test_dir, flush_millis=10000, name='test') test_dir, flush_millis=10000, name='test')
checkpoint_prefix = os.path.join(FLAGS.checkpoint_dir, 'ckpt') checkpoint_prefix = os.path.join(FLAGS.checkpoint_dir, 'ckpt')
# Train and evaluate for 11 epochs.
with tf.device(device): with tf.device(device):
for epoch in range(1, 11): for epoch in range(1, 11):
with tfe.restore_variables_on_create( with tfe.restore_variables_on_create(
......
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