We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents ae7fb2a + 5b5fa37 commit ae0d806Copy full SHA for ae0d806
python/paddle/fluid/tests/unittests/test_reader_reset.py
@@ -13,7 +13,7 @@
13
# limitations under the License.
14
15
from __future__ import print_function
16
-
+import os
17
import paddle.fluid as fluid
18
import paddle
19
import numpy as np
@@ -41,6 +41,8 @@ def fake_data_generator():
41
self.data_file_name, reader, feeder)
42
43
def setUp(self):
44
+ # set parallel threads to fit 20 batches in line 49
45
+ os.environ['CPU_NUM'] = str(20)
46
self.use_cuda = fluid.core.is_compiled_with_cuda()
47
self.data_file_name = './reader_reset_test.recordio'
48
self.ins_shape = [3]
0 commit comments