Browse Source

change resnet101 allreduce split

pull/13875/head
zhaoting 4 years ago
parent
commit
ac76deb3da
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/official/cv/resnet/train.py

+ 1
- 1
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:


Loading…
Cancel
Save