From b8a659ac4c689f316c8bb5f609d83702fe9da599 Mon Sep 17 00:00:00 2001 From: Benjamin Herta Date: Wed, 24 Jan 2018 16:40:36 -0500 Subject: [PATCH] Remove unnecessary call to glob --- scripts/tf_cnn_benchmarks/preprocessing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/tf_cnn_benchmarks/preprocessing.py b/scripts/tf_cnn_benchmarks/preprocessing.py index 6213e5c1..bfdf08dd 100644 --- a/scripts/tf_cnn_benchmarks/preprocessing.py +++ b/scripts/tf_cnn_benchmarks/preprocessing.py @@ -503,7 +503,7 @@ def minibatch(self, dataset, subset, use_datasets, cache_data, if not file_names: raise ValueError('Found no files in --data_dir matching: {}' .format(glob_pattern)) - ds = tf.data.TFRecordDataset.list_files(file_names) + ds = tf.data.TFRecordDataset.list_files(glob_pattern) ds = ds.apply( interleave_ops.parallel_interleave( tf.data.TFRecordDataset, cycle_length=10))