未验证 提交 2611ed25 编写于 作者: C Chen Weihang 提交者: GitHub

Polish unittest test_decoupled_py_reader (#32832)

* polish test decoupled py reader

* remove bin file
上级 1ef23279
...@@ -19,9 +19,9 @@ import time ...@@ -19,9 +19,9 @@ import time
import six import six
import unittest import unittest
EPOCH_NUM = 20 EPOCH_NUM = 5
BATCH_SIZE = 32 BATCH_SIZE = 16
BATCH_NUM = 20 BATCH_NUM = 10
CLASS_NUM = 10 CLASS_NUM = 10
...@@ -29,7 +29,7 @@ def random_reader(): ...@@ -29,7 +29,7 @@ def random_reader():
np.random.seed(1) np.random.seed(1)
for i in range(BATCH_SIZE * BATCH_NUM): for i in range(BATCH_SIZE * BATCH_NUM):
image = np.random.random([784]) 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 yield image, label
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册