Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

del ccl param in config.py

This commit is contained in:
zhaoting 2020-11-24 17:11:11 +08:00
parent 5b9ab437dd
commit 28bdda2bb3
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/mobilenetv2/src/config.py
View File

@ -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')),

2
model_zoo/official/cv/mobilenetv2/src/utils.py
View File

@ -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)