未验证 提交 ab1a6cb8 编写于 作者: Y yuyang18

Remove warning

上级 39231804
...@@ -142,8 +142,7 @@ class TestDataBalance(unittest.TestCase): ...@@ -142,8 +142,7 @@ class TestDataBalance(unittest.TestCase):
filenames=[self.lod_data_file_name], filenames=[self.lod_data_file_name],
shapes=[[-1, 3], [-1, 1]], shapes=[[-1, 3], [-1, 1]],
lod_levels=[1, 0], lod_levels=[1, 0],
dtypes=['float32', 'int32'], dtypes=['float32', 'int32'])
thread_num=1)
ins, label = fluid.layers.read_file(data_reader) ins, label = fluid.layers.read_file(data_reader)
place = fluid.CUDAPlace(0) if self.use_cuda else fluid.CPUPlace() place = fluid.CUDAPlace(0) if self.use_cuda else fluid.CPUPlace()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册