未验证 提交 8c922931 编写于 作者: C Chen Weihang 提交者: GitHub

reduce iter time to reduce exec time (#26388)

上级 4ad504e7
...@@ -34,7 +34,7 @@ class TestParallelExecutorBase(unittest.TestCase): ...@@ -34,7 +34,7 @@ class TestParallelExecutorBase(unittest.TestCase):
def check_network_convergence(cls, def check_network_convergence(cls,
method, method,
use_cuda=True, use_cuda=True,
iter=50, iter=5,
batch_size=None, batch_size=None,
feed_dict=None, feed_dict=None,
feed_data_reader=None, feed_data_reader=None,
......
...@@ -179,7 +179,7 @@ def batch_size(use_cuda): ...@@ -179,7 +179,7 @@ def batch_size(use_cuda):
def iter(use_cuda): def iter(use_cuda):
if use_cuda: if use_cuda:
return 10 return 10
return 2 return 1
gpu_img, gpu_label = init_data( gpu_img, gpu_label = init_data(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册