提交 a944d571 编写于 作者: F fengjiayi

refine code

上级 4286ea61
......@@ -61,6 +61,5 @@ class TestMultipleReader(unittest.TestCase):
img_val, = exe.run(fetch_list=[img])
batch_count += 1
self.assertLessEqual(img_val.shape[0], self.batch_size)
print(batch_count)
data_file.reset()
self.assertEqual(batch_count, self.num_batch * self.pass_num)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册