Skip to content

Commit c6fc6a3

Browse files
committed
fix a bugs
1 parent 0dff1e4 commit c6fc6a3

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

python/paddle/fluid/tests/unittests/test_multi_file_reader.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,6 @@ def main(self, thread_num):
6969
break
7070
batch_count += 1
7171
self.assertLessEqual(img_val.shape[0], self.batch_size)
72-
data_files.reset()
7372
self.assertEqual(batch_count, self.num_batch * 3)
7473

7574
def test_main(self):

python/paddle/fluid/tests/unittests/test_recordio_reader.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,13 +74,13 @@ def test_main(self, decorator_callback=None):
7474

7575
avg_loss_np.append(tmp)
7676
batch_id += 1
77-
data_file.reset()
7877
self.assertEqual(batch_id, self.num_batches)
7978
self.assertLess(avg_loss_np[-1], avg_loss_np[0])
8079

8180
def test_shuffle_reader(self):
82-
self.test_main(decorator_callback=lambda reader: fluid.layers.io.shuffle(reader, buffer_size=200))
81+
self.test_main(decorator_callback=lambda reader: fluid.layers.io.shuffle(
82+
reader, buffer_size=200))
8383

8484
def test_double_buffer_reader(self):
8585
self.test_main(decorator_callback=lambda reader: fluid.layers.io.double_buffer(reader,
86-
place='cuda:0' if fluid.core.is_compiled_with_cuda() else 'cpu'))
86+
place='cuda:0' if fluid.core.is_compiled_with_cuda() else 'cpu'))

0 commit comments

Comments
 (0)