From 2e788bd81e2720cdb1beceaa0f9e73996e2c0ac1 Mon Sep 17 00:00:00 2001 From: Huihuang Zheng Date: Fri, 18 Dec 2020 10:55:59 +0800 Subject: [PATCH] Reduce batch size ot fix CPU memory, test=develop (#29736) Unit test reported memory not enough on CPU machines. Reduce batch size again. --- 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 d8f088019ba..ff4ce0f3231 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/yolov3.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/yolov3.py @@ -93,7 +93,7 @@ cfg.ignore_thresh = .7 # SOLVER options # # batch size -cfg.batch_size = 2 if sys.platform == 'darwin' or os.name == 'nt' else 4 +cfg.batch_size = 1 if sys.platform == 'darwin' or os.name == 'nt' else 4 # derived learning rate the to get the final learning rate. cfg.learning_rate = 0.001 # maximum number of iterations -- GitLab