diff --git a/fluid/PaddleCV/human_pose_estimation/train.py b/fluid/PaddleCV/human_pose_estimation/train.py index 10666310e8a849d5ee26df1c0a7d52826210bb73..58e2b39ee6af389f1da0a2fb838e3cbac7597f8c 100644 --- a/fluid/PaddleCV/human_pose_estimation/train.py +++ b/fluid/PaddleCV/human_pose_estimation/train.py @@ -45,6 +45,8 @@ add_arg('lr', float, 0.001, "Set learning rate.") add_arg('lr_strategy', str, "piecewise_decay", "Set the learning rate decay strategy.") parser.add_argument('--enable_ce', action='store_true', help='If set, run the task with continuous evaluation logs.') parser.add_argument('--batch_num', type=int, help="batch num for ce") +parser.add_argument('--num_devices', type=int, default=1, help='Number of GPU devices') + # yapf: enable def optimizer_setting(args, params):