p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!13875 change resnet101 allreduce split 

From: @zhao_ting_v
Reviewed-by: @wuxuejian,@oacjiewen
Signed-off-by: @oacjiewen
This commit is contained in:
mindspore-ci-bot 2021-03-24 20:16:35 +08:00 committed by Gitee
parent be0e2f18e6 ac76deb3da
commit 1b136e1713
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/resnet/train.py
View File

@ -111,7 +111,7 @@ if __name__ == '__main__':
if args_opt.net == "resnet50" or args_opt.net == "se-resnet50":
context.set_auto_parallel_context(all_reduce_fusion_config=[85, 160])
elif args_opt.net == "resnet101":
context.set_auto_parallel_context(all_reduce_fusion_config=[180, 313])
context.set_auto_parallel_context(all_reduce_fusion_config=[80, 210, 313])
init()
# GPU target
else: