reduce py_reader unittest time (#16996)

test=develop
6 years ago · af8a041bb6
parent 45ebafc33e
commit af8a041bb6
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/test_decoupled_py_reader.py
+++ b/python/paddle/fluid/tests/unittests/test_decoupled_py_reader.py
@ -21,12 +21,13 @@ import unittest

 EPOCH_NUM = 20
 BATCH_SIZE = 32
+BATCH_NUM = 20
 CLASS_NUM = 10


 def random_reader():
    np.random.seed(1)
-    for i in range(BATCH_SIZE * 40):
+    for i in range(BATCH_SIZE * BATCH_NUM):
        image = np.random.random([784])
        label = np.random.random_integers(low=0, high=CLASS_NUM - 1)
        yield image, label