From 46d16006576934680afad03ca84953db362f3374 Mon Sep 17 00:00:00 2001 From: Payne Date: Sat, 10 Oct 2020 17:38:24 +0800 Subject: [PATCH] update context api args --- model_zoo/official/cv/mobilenetv2/src/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.")