Skip to content

GitLab

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