Skip to content

Commit 7b5fac8

Browse files
committed
fix
1 parent 468ce6c commit 7b5fac8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: python/data.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -84,7 +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(file_name, num_epochs=max_epoch)
87+
filename_queue = tf.train.string_input_producer(tf.train.match_filenames_once(file_name), num_epochs=max_epoch)
8888
serialized_example = self.Decode(filename_queue)
8989
capacity = thread_num * batch_size + min_after_dequeue
9090
batch_serialized_example = tf.train.shuffle_batch(

0 commit comments

Comments
 (0)