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

update context api args

This commit is contained in:
Payne 2020-10-10 17:38:24 +08:00
parent ae06c726ca
commit 46d1600657
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -49,7 +49,7 @@ def context_device_init(config):
if config.run_distribute: if config.run_distribute:
context.set_auto_parallel_context(device_num=config.rank_size, context.set_auto_parallel_context(device_num=config.rank_size,
parallel_mode=ParallelMode.DATA_PARALLEL, parallel_mode=ParallelMode.DATA_PARALLEL,
gradients_mean=True, set_all_reduce_fusion_split_indices=[140]) gradients_mean=True, all_reduce_fusion_config=[140])
init() init()
else: else:
raise ValueError("Only support CPU, GPU and Ascend.") raise ValueError("Only support CPU, GPU and Ascend.")