提交 a87e666e 编写于 作者: C caojian05

remove enable_hccl setting in context for vgg16

上级 2c85295e
...@@ -71,7 +71,6 @@ if __name__ == '__main__': ...@@ -71,7 +71,6 @@ if __name__ == '__main__':
device_num = int(os.environ.get("DEVICE_NUM", 1)) device_num = int(os.environ.get("DEVICE_NUM", 1))
if device_num > 1: if device_num > 1:
context.reset_auto_parallel_context() context.reset_auto_parallel_context()
context.set_context(enable_hccl=True)
context.set_auto_parallel_context(device_num=device_num, parallel_mode=ParallelMode.DATA_PARALLEL, context.set_auto_parallel_context(device_num=device_num, parallel_mode=ParallelMode.DATA_PARALLEL,
mirror_mean=True) mirror_mean=True)
init() init()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册