From 6482053e12a8c06a2d49fc44542ca2727984c952 Mon Sep 17 00:00:00 2001 From: gengdongjie Date: Thu, 30 Apr 2020 16:17:06 +0800 Subject: [PATCH] remove enable_hccl option in context --- chapter05/resnet/resnet_cifar.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/chapter05/resnet/resnet_cifar.py b/chapter05/resnet/resnet_cifar.py index 2424f88..c3ea22a 100644 --- a/chapter05/resnet/resnet_cifar.py +++ b/chapter05/resnet/resnet_cifar.py @@ -114,16 +114,10 @@ def create_dataset(repeat_num=1, training=True): return ds if __name__ == '__main__': - if args_opt.do_eval: - context.set_context(enable_hccl=False) - else: - if args_opt.run_distribute: - context.set_context(enable_hccl=True) - context.set_auto_parallel_context(device_num=args_opt.device_num, parallel_mode=ParallelMode.DATA_PARALLEL, mirror_mean=True) - auto_parallel_context().set_all_reduce_fusion_split_indices([140]) - init() - else: - context.set_context(enable_hccl=False) + if not args_opt.do_eval and args_opt.run_distribute: + context.set_auto_parallel_context(device_num=args_opt.device_num, parallel_mode=ParallelMode.DATA_PARALLEL, mirror_mean=True) + auto_parallel_context().set_all_reduce_fusion_split_indices([140]) + init() epoch_size = args_opt.epoch_size net = resnet50(args_opt.num_classes) -- GitLab