Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!9454 fix hccl init in resnext50 

From: @zhouyaqiang0
Reviewed-by: @c_34,@guoqi1024
Signed-off-by: @c_34
This commit is contained in:
mindspore-ci-bot 2020-12-04 10:09:02 +08:00 committed by Gitee
parent f019a4a0af 6fbb124b1d
commit a6766001ec
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_zoo/official/cv/resnext50/train.py
View File

@ -147,6 +147,8 @@ def parse_args(cloud_args=None):
args.lr_epochs = list(map(int, args.lr_epochs.split(',')))
args.image_size = list(map(int, args.image_size.split(',')))
context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True,
device_target=args.platform, save_graphs=False)
# init distributed
if args.is_distributed:
init()
@ -190,8 +192,6 @@ def merge_args(args, cloud_args):
def train(cloud_args=None):
"""training process"""
args = parse_args(cloud_args)
context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True,
device_target=args.platform, save_graphs=False)
if os.getenv('DEVICE_ID', "not_set").isdigit():
context.set_context(device_id=int(os.getenv('DEVICE_ID')))