From 2e5e526e550766f0db0fc88c922f8304e604f698 Mon Sep 17 00:00:00 2001 From: lichenever Date: Thu, 10 Sep 2020 19:47:04 +0800 Subject: [PATCH] fix_model_zool_resnet50_script_bug --- model_zoo/official/cv/resnet/train.py | 2 +- model_zoo/official/cv/resnet_thor/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/cv/resnet/train.py b/model_zoo/official/cv/resnet/train.py index 0cc619b9bad..d4572b93fc7 100755 --- a/model_zoo/official/cv/resnet/train.py +++ b/model_zoo/official/cv/resnet/train.py @@ -78,7 +78,7 @@ if __name__ == '__main__': context.set_auto_parallel_context(device_num=args_opt.device_num, parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True) if args_opt.net == "resnet50" or args_opt.net == "se-resnet50": - context.set_auto_parallel_context(all_reduce_fusion_config=[85, 150]) + context.set_auto_parallel_context(all_reduce_fusion_config=[85, 160]) else: context.set_auto_parallel_context(all_reduce_fusion_config=[180, 313]) init() diff --git a/model_zoo/official/cv/resnet_thor/train.py b/model_zoo/official/cv/resnet_thor/train.py index 7de1034628b..1954fbfde21 100644 --- a/model_zoo/official/cv/resnet_thor/train.py +++ b/model_zoo/official/cv/resnet_thor/train.py @@ -99,7 +99,7 @@ if __name__ == '__main__': else: init() context.set_auto_parallel_context(device_num=get_group_size(), parallel_mode=ParallelMode.DATA_PARALLEL, - gradients_mean=True, all_reduce_fusion_config=[104]) + gradients_mean=True, all_reduce_fusion_config=[107]) ckpt_save_dir = config.save_checkpoint_path + "ckpt_" + str(get_rank()) + "/" # create dataset