diff --git a/official/cv/Deepsort/generater-detection.py b/official/cv/Deepsort/generater-detection.py index 5173c16aecacd9fb3ca47834a24d71bda5b088ac..83e4c4e6b227eeef58af91250ebd8f7ff581710a 100644 --- a/official/cv/Deepsort/generater-detection.py +++ b/official/cv/Deepsort/generater-detection.py @@ -212,7 +212,7 @@ def parse_args(): if __name__ == "__main__": - context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True, + context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", save_graphs=False) args = parse_args() if args.run_modelarts: diff --git a/official/cv/brdnet/eval.py b/official/cv/brdnet/eval.py index 253e1c1aad817a3733f3f8cecdaa38737122b58a..0573f2df969ae698d3c482da2d253ab9e741ae54 100644 --- a/official/cv/brdnet/eval.py +++ b/official/cv/brdnet/eval.py @@ -172,7 +172,7 @@ def test(model_path): if __name__ == '__main__': device_id = int(os.getenv('DEVICE_ID', '0')) - context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True, + context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target, device_id=device_id, save_graphs=False) args.logger = get_logger(save_dir, "BRDNet", 0) diff --git a/official/cv/brdnet/train.py b/official/cv/brdnet/train.py index b83fec8deef1b39d5d156bea62425915bd86b61a..9e8d1f97cca2ef7c0c3238911af71fa2d1d876f3 100644 --- a/official/cv/brdnet/train.py +++ b/official/cv/brdnet/train.py @@ -86,7 +86,7 @@ def get_lr(steps_per_epoch, max_epoch, init_lr): device_id = int(os.getenv('DEVICE_ID', '0')) -context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True, +context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target, save_graphs=False) def train(): diff --git a/official/cv/resnet/modelarts/ResNet18/modelarts_train.py b/official/cv/resnet/modelarts/ResNet18/modelarts_train.py index 0abf4aa3dd2bde6f62ffd770e0ebb04eef335d88..9bd19d6eaf24df6b50c7b5c057505f78cc8ef816 100644 --- a/official/cv/resnet/modelarts/ResNet18/modelarts_train.py +++ b/official/cv/resnet/modelarts/ResNet18/modelarts_train.py @@ -202,8 +202,7 @@ def init_context(target): 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_context(device_id=device_id) context.set_auto_parallel_context( device_num=args_opt.device_num, parallel_mode=ParallelMode.DATA_PARALLEL, diff --git a/official/cv/se_resnext50/eval.py b/official/cv/se_resnext50/eval.py index e339710b4b4400cc96fb32033b0dc5109002385d..7cbde15eb143cf642a8a4f31b7e5ebe6b91285f8 100644 --- a/official/cv/se_resnext50/eval.py +++ b/official/cv/se_resnext50/eval.py @@ -124,7 +124,7 @@ def get_result(model, top1_correct, top5_correct, img_tot): def test(): """test""" set_parameters() - context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True, + context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target, save_graphs=False) if os.getenv('DEVICE_ID', "not_set").isdigit(): context.set_context(device_id=int(os.getenv('DEVICE_ID'))) diff --git a/official/cv/se_resnext50/train.py b/official/cv/se_resnext50/train.py index 97ebb71fc89d67e96e69fd075e082e17a7e20c27..ee0cd354ec62733da0fd30159a367ca50183c626 100644 --- a/official/cv/se_resnext50/train.py +++ b/official/cv/se_resnext50/train.py @@ -110,7 +110,7 @@ class ProgressMonitor(Callback): def set_parameters(): """parameters""" - context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True, + context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target, save_graphs=False) # init distributed if config.run_distribute: diff --git a/official/cv/yolov5/train.py b/official/cv/yolov5/train.py index e3ef1b8a4b982a33710ce3742b3cc45ca5a4785f..c2130510d7eca40ef2dd07c4d19a049e4660363a 100644 --- a/official/cv/yolov5/train.py +++ b/official/cv/yolov5/train.py @@ -54,7 +54,7 @@ def set_default(): outputs_dir = config.ckpt_path device_id = get_device_id() - context.set_context(mode=context.GRAPH_MODE, enable_auto_mixed_precision=True, device_target=config.device_target, + context.set_context(mode=context.GRAPH_MODE, device_target=config.device_target, save_graphs=False, device_id=device_id) # init distributed if config.is_distributed: diff --git a/research/audio/fcn-4/train.py b/research/audio/fcn-4/train.py index cedb943523b89de0fa7de3c7385acfca095cc3bb..2c1fda7414cb1edbd24f123fba528fa9a889099c 100644 --- a/research/audio/fcn-4/train.py +++ b/research/audio/fcn-4/train.py @@ -61,7 +61,6 @@ if __name__ == "__main__": config.checkpoint_path = os.path.abspath(config.checkpoint_path) context.set_context(device_target=config.device_target, mode=context.GRAPH_MODE) - context.set_context(enable_auto_mixed_precision=config.mixed_precision) if config.device_target == 'Ascend': context.set_context(device_id=get_device_id()) diff --git a/research/cv/glore_res/train.py b/research/cv/glore_res/train.py index 6c5015f022159ed59d3bc07f296dac81e80bd32c..71dfa416934d6263b80c5e65e4739094963ec28e 100644 --- a/research/cv/glore_res/train.py +++ b/research/cv/glore_res/train.py @@ -65,7 +65,7 @@ if __name__ == '__main__': if config.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_context(device_id=device_id) if config.net == 'resnet200': context.set_auto_parallel_context(parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True,