!13875 change resnet101 allreduce split

From: @zhao_ting_v Reviewed-by: @wuxuejian,@oacjiewen Signed-off-by: @oacjiewen
4 years ago · 1b136e1713
parent be0e2f18e6 ac76deb3da
commit 1b136e1713
1 changed files with 1 additions and 1 deletions
--- a/model_zoo/official/cv/resnet/train.py
+++ b/model_zoo/official/cv/resnet/train.py
@ -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: