From 28d074e95655e6c1fe74c9a3abcdda5859b66d70 Mon Sep 17 00:00:00 2001 From: Huihuang Zheng Date: Wed, 10 Jun 2020 14:05:36 +0800 Subject: [PATCH] [Dy2stat] Decrease test_yolov3 GPU usage (#24955) [Dy2stat] decrease the batch size to decrease GPU usage. --- python/paddle/fluid/tests/unittests/dygraph_to_static/yolov3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/yolov3.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/yolov3.py index 2740f7ef28..3f55af9647 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/yolov3.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/yolov3.py @@ -90,7 +90,7 @@ cfg.ignore_thresh = .7 # SOLVER options # # 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. cfg.learning_rate = 0.001 # maximum number of iterations -- GitLab