!7167 master context parameter update

Merge pull request !7167 from yepei6/master_modify_context
4 years ago · 84f3a9531e
parent 7af0d3374f 46d1600657
commit 84f3a9531e
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/mobilenetv2/src/utils.py
+++ b/model_zoo/official/cv/mobilenetv2/src/utils.py
@ -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.")