diff --git a/model_zoo/official/cv/mobilenetv2/src/config.py b/model_zoo/official/cv/mobilenetv2/src/config.py index 5eeceeec51..9598ca3310 100644 --- a/model_zoo/official/cv/mobilenetv2/src/config.py +++ b/model_zoo/official/cv/mobilenetv2/src/config.py @@ -63,7 +63,6 @@ def set_config(args): "keep_checkpoint_max": 200, "save_checkpoint_path": "./", "platform": args.platform, - "ccl": "nccl", "run_distribute": args.run_distribute, "activation": "Softmax", "export_format": "MINDIR", @@ -88,7 +87,6 @@ def set_config(args): "keep_checkpoint_max": 200, "save_checkpoint_path": "./", "platform": args.platform, - "ccl": "hccl", "device_id": int(os.getenv('DEVICE_ID', '0')), "rank_id": int(os.getenv('RANK_ID', '0')), "rank_size": int(os.getenv('RANK_SIZE', '1')), diff --git a/model_zoo/official/cv/mobilenetv2/src/utils.py b/model_zoo/official/cv/mobilenetv2/src/utils.py index 586a4d21bf..f411c76d21 100644 --- a/model_zoo/official/cv/mobilenetv2/src/utils.py +++ b/model_zoo/official/cv/mobilenetv2/src/utils.py @@ -38,7 +38,7 @@ def context_device_init(config): elif config.platform == "GPU": context.set_context(mode=context.GRAPH_MODE, device_target=config.platform, save_graphs=False) if config.run_distribute: - init("nccl") + init() context.set_auto_parallel_context(device_num=get_group_size(), parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True)