From 9efc2e4eb259d5102e00af5ee1cc6313d004f588 Mon Sep 17 00:00:00 2001 From: chenhaozhe Date: Mon, 26 Apr 2021 11:31:17 +0800 Subject: [PATCH] fix mobilnetv1 1p error: device_id not defined --- model_zoo/official/cv/mobilenetv1/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/mobilenetv1/train.py b/model_zoo/official/cv/mobilenetv1/train.py index 8da7bed47a..32f5452c95 100755 --- a/model_zoo/official/cv/mobilenetv1/train.py +++ b/model_zoo/official/cv/mobilenetv1/train.py @@ -61,9 +61,9 @@ 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') if args_opt.run_distribute: if target == "Ascend": - device_id = int(os.getenv('DEVICE_ID')) context.set_context(device_id=device_id, enable_auto_mixed_precision=True) context.set_auto_parallel_context(device_num=args_opt.device_num, parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True)