未验证 提交 01294ff9 编写于 作者: H Huihuang Zheng 提交者: GitHub

Decrease the batch size of se-resnext in unittest. (#24651)

As discussed with QA, we will use p4 machine for unit test and the GPU on those machine may not have enough GPU, which can cause "test_parallel_executor_seresnext_base_gpu" failed. So I decrease the batch size.
上级 21ad122a
...@@ -173,7 +173,7 @@ model = SE_ResNeXt50Small ...@@ -173,7 +173,7 @@ model = SE_ResNeXt50Small
def batch_size(): def batch_size():
return 12 return 8
def iter(use_cuda): def iter(use_cuda):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册