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

fix env error

This commit is contained in:
chenhaozhe 2021-04-27 10:57:11 +08:00
parent 55bdb33f35
commit 122378549a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/mobilenetv1/train.py
View File

@ -61,7 +61,7 @@ if __name__ == '__main__':
context.set_context(mode=context.GRAPH_MODE, device_target=target, save_graphs=False)
if args_opt.parameter_server:
context.set_ps_context(enable_ps=True)
device_id = int(os.getenv('DEVICE_ID'), '0')
device_id = int(os.getenv('DEVICE_ID', '0'))
if args_opt.run_distribute:
if target == "Ascend":
context.set_context(device_id=device_id, enable_auto_mixed_precision=True)