From eea7090c26a1ac8bf559f76e1f04e950c5aef1f6 Mon Sep 17 00:00:00 2001 From: gongweibao Date: Tue, 5 Jan 2021 13:47:42 +0800 Subject: [PATCH] fix selected_gpus test=develop (#30044) --- python/paddle/distributed/fleet/launch.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/paddle/distributed/fleet/launch.py b/python/paddle/distributed/fleet/launch.py index afc352f89c..c7c60a3fbd 100644 --- a/python/paddle/distributed/fleet/launch.py +++ b/python/paddle/distributed/fleet/launch.py @@ -117,6 +117,8 @@ see: http://www.paddlepaddle.org/documentation/docs/zh/1.6/user_guides/howto/tra "--gpus=\"0,1,2,3\" will launch four training processes each bound to one gpu." ) + base_group.add_argument("--selected_gpus", dest="gpus") + base_group.add_argument( "training_script", type=str, -- GitLab