update context api args

5 years ago · 46d1600657
parent ae06c726ca
commit 46d1600657
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.")