diff --git a/model_zoo/official/cv/mobilenetv2/src/utils.py b/model_zoo/official/cv/mobilenetv2/src/utils.py
index 9c6451f6f67..fcec3562bc4 100644
--- 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.")