Polish unittest test_decoupled_py_reader (#32832)

* polish test decoupled py reader * remove bin file
PaddlePaddle · May 12, 2021 · 2611ed2 · 2611ed2
1 parent 1ef2327
commit 2611ed2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_decoupled_py_reader.py b/python/paddle/fluid/tests/unittests/test_decoupled_py_reader.py
@@ -19,17 +19,17 @@
 import six
 import unittest
 
-EPOCH_NUM = 20
-BATCH_SIZE = 32
-BATCH_NUM = 20
+EPOCH_NUM = 5
+BATCH_SIZE = 16
+BATCH_NUM = 10
 CLASS_NUM = 10
 
 
 def random_reader():
     np.random.seed(1)
     for i in range(BATCH_SIZE * BATCH_NUM):
         image = np.random.random([784])
-        label = np.random.random_integers(low=0, high=CLASS_NUM - 1)
+        label = np.random.randint(low=0, high=CLASS_NUM)
         yield image, label