Browse Source

update context api args

tags/v1.1.0
Payne 5 years ago
parent
commit
46d1600657
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/mobilenetv2/src/utils.py

+ 1
- 1
model_zoo/official/cv/mobilenetv2/src/utils.py View File

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


Loading…
Cancel
Save