!12157 fix mobilenetv3 gpu train param

From: @wuxuejian Reviewed-by: @liangchenghui,@c_34 Signed-off-by: @liangchenghui
4 years ago · 408159e301
parent 2d2e824b77 3d185352e0
commit 408159e301
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/mobilenetv3/train.py
+++ b/model_zoo/official/cv/mobilenetv3/train.py
@ -185,7 +185,7 @@ if __name__ == '__main__':
                                 device_target=args_opt.device_target,
                                 repeat_num=1,
                                 batch_size=config_.batch_size,
-                                 run_distribute=False)
+                                 run_distribute=args_opt.run_distribute)
    elif args_opt.device_target == "CPU":
        dataset = create_dataset_cifar(args_opt.dataset_path,
                                       do_train=True,