Skip to content

GitLab

Commit 70c86f24 authored by Changming Sun's avatar Changming Sun Committed by Neal Wu
Browse files

make official/mnist support hdfs input (#2453)

parent 2d1e1782
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/mnist/mnist.py
View file @ 70c86f24
......@@ -64,8 +64,8 @@ def input_fn(mode, batch_size=1):
assert mode == tf.estimator.ModeKeys.EVAL, 'invalid mode'
tfrecords_file = os.path.join(FLAGS.data_dir, 'test.tfrecords')
assert os.path.exists(tfrecords_file), ('Run convert_to_records.py first to '
'convert the MNIST data to TFRecord file format.')
assert tf.gfile.Exists(tfrecords_file), ('Run convert_to_records.py first to '
'convert the MNIST data to TFRecord file format.')
dataset = tf.contrib.data.TFRecordDataset([tfrecords_file])
......
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