Merge pull request #501 from qsbao/list_files_no_shuffle

list_files in inference examples should be deterministically

Author: leewyang

examples/mnist/estimator/mnist_inference.py

@@ -47,7 +47,7 @@ def parse_tfr(example_proto):
     return (image, label)
 
   # define a new tf.data.Dataset (for inferencing)
-  ds = tf.data.Dataset.list_files("{}/part-*".format(args.images_labels))
+  ds = tf.data.Dataset.list_files("{}/part-*".format(args.images_labels), shuffle=False)
   ds = ds.shard(num_workers, worker_num)
   ds = ds.interleave(tf.data.TFRecordDataset)
   ds = ds.map(parse_tfr)

examples/mnist/keras/mnist_inference.py

@@ -38,7 +38,7 @@ def parse_tfr(example_proto):
     return (image, label)
 
   # define a new tf.data.Dataset (for inferencing)
-  ds = tf.data.Dataset.list_files("{}/part-*".format(args.images_labels))
+  ds = tf.data.Dataset.list_files("{}/part-*".format(args.images_labels), shuffle=False)
   ds = ds.shard(ctx.num_workers, ctx.worker_num)
   ds = ds.interleave(tf.data.TFRecordDataset)
   ds = ds.map(parse_tfr)