fix mac test_reader_reset

7 years ago · 415e0eac69
parent f6cbe10a80
commit 415e0eac69
1 changed files with 2 additions and 0 deletions
--- a/python/paddle/fluid/tests/unittests/test_reader_reset.py
+++ b/python/paddle/fluid/tests/unittests/test_reader_reset.py
@ -41,6 +41,8 @@ class TestReaderReset(unittest.TestCase):
                self.data_file_name, reader, feeder)

    def setUp(self):
+        # set parallel threads to fit 20 batches in line 49
+        os.environ['CPU_NUM'] = str(20)
        self.use_cuda = fluid.core.is_compiled_with_cuda()
        self.data_file_name = './reader_reset_test.recordio'
        self.ins_shape = [3]