We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 468ce6c commit 7b5fac8Copy full SHA for 7b5fac8
python/data.py
@@ -84,7 +84,7 @@ def ReadBatch(self, file_name, max_epoch, batch_size, thread_num, min_after_dequ
84
Return Tensor and SparseTensor parsed from tfrecord
85
'''
86
with tf.name_scope('input'):
87
- filename_queue = tf.train.string_input_producer(file_name, num_epochs=max_epoch)
+ filename_queue = tf.train.string_input_producer(tf.train.match_filenames_once(file_name), num_epochs=max_epoch)
88
serialized_example = self.Decode(filename_queue)
89
capacity = thread_num * batch_size + min_after_dequeue
90
batch_serialized_example = tf.train.shuffle_batch(
0 commit comments