fix mobilenetv3 gpu train param

4 years ago · 3d185352e0
parent ca3f916c1e
commit 3d185352e0
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,