diff --git a/python/paddle/distributed/fleet/launch.py b/python/paddle/distributed/fleet/launch.py index afc352f89cba656d7b618faf81549dfe18cc8c31..c7c60a3fbde063d2fd6184e85a8b256f5b3a40a1 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,