!8957 delete redundant param ccl in config.py

From: @zhao_ting_v
Reviewed-by: @liangchenghui,@linqingke
Signed-off-by: @liangchenghui
pull/8957/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 9f819cd303

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

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

Loading…
Cancel
Save