Skip to content

Commit 468ce6c

Browse files
committed
fix
1 parent 2a9d4a8 commit 468ce6c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

python/data.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,7 @@ def ReadBatch(self, file_name, max_epoch, batch_size, thread_num, min_after_dequ
8484
Return Tensor and SparseTensor parsed from tfrecord
8585
'''
8686
with tf.name_scope('input'):
87-
filename_queue = tf.train.string_input_producer(
88-
tf.train.match_filenames_once(file_name), num_epochs=max_epoch)
87+
filename_queue = tf.train.string_input_producer(file_name, num_epochs=max_epoch)
8988
serialized_example = self.Decode(filename_queue)
9089
capacity = thread_num * batch_size + min_after_dequeue
9190
batch_serialized_example = tf.train.shuffle_batch(

0 commit comments

Comments
 (0)