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 d8f088019ba464c7c7f5075c734371848c24536b..ff4ce0f32319a19aeb9fc52601ef1be2ff706f44 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