未验证 提交 28d074e9 编写于 作者: H Huihuang Zheng 提交者: GitHub

[Dy2stat] Decrease test_yolov3 GPU usage (#24955)

[Dy2stat] decrease the batch size to decrease GPU usage.
上级 ff8ca52f
...@@ -90,7 +90,7 @@ cfg.ignore_thresh = .7 ...@@ -90,7 +90,7 @@ cfg.ignore_thresh = .7
# SOLVER options # SOLVER options
# #
# batch size # batch size
cfg.batch_size = 8 if fluid.is_compiled_with_cuda() else 4 cfg.batch_size = 4
# derived learning rate the to get the final learning rate. # derived learning rate the to get the final learning rate.
cfg.learning_rate = 0.001 cfg.learning_rate = 0.001
# maximum number of iterations # maximum number of iterations
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册