From ab1a6cb8b84c8574194b4b57e217fa8658b9be7a Mon Sep 17 00:00:00 2001 From: yuyang18 Date: Sat, 14 Jul 2018 12:06:27 +0800 Subject: [PATCH] Remove warning --- python/paddle/fluid/tests/unittests/test_data_balance.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_data_balance.py b/python/paddle/fluid/tests/unittests/test_data_balance.py index 023d31cc8..7fa4abc61 100644 --- a/python/paddle/fluid/tests/unittests/test_data_balance.py +++ b/python/paddle/fluid/tests/unittests/test_data_balance.py @@ -142,8 +142,7 @@ class TestDataBalance(unittest.TestCase): filenames=[self.lod_data_file_name], shapes=[[-1, 3], [-1, 1]], lod_levels=[1, 0], - dtypes=['float32', 'int32'], - thread_num=1) + dtypes=['float32', 'int32']) ins, label = fluid.layers.read_file(data_reader) place = fluid.CUDAPlace(0) if self.use_cuda else fluid.CPUPlace() -- GitLab