未验证 提交 ac2010f4 编写于 作者: L liym27 提交者: GitHub

[Fix test_resnet timeout] Reduce batch_size to 2 to avoid unittest timeout.(#26449)

上级 fd66d762
......@@ -31,7 +31,8 @@ IMAGENET1000 = 1281167
base_lr = 0.001
momentum_rate = 0.9
l2_decay = 1e-4
batch_size = 8
# NOTE: Reduce batch_size from 8 to 2 to avoid unittest timeout.
batch_size = 2
epoch_num = 1
place = fluid.CUDAPlace(0) if fluid.is_compiled_with_cuda() \
else fluid.CPUPlace()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册