From 7dbacd0ddf9291c065e43f74da5d7635d327ab7a Mon Sep 17 00:00:00 2001 From: PingqiLi Date: Tue, 19 Mar 2024 11:41:44 +0800 Subject: [PATCH 1/2] fix the config of squeezenet_1.0 with amp_level O0 --- configs/squeezenet/squeezenet_1.0_ascend.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/squeezenet/squeezenet_1.0_ascend.yaml b/configs/squeezenet/squeezenet_1.0_ascend.yaml index aed5d70b..79c06bbd 100644 --- a/configs/squeezenet/squeezenet_1.0_ascend.yaml +++ b/configs/squeezenet/squeezenet_1.0_ascend.yaml @@ -30,7 +30,7 @@ keep_checkpoint_max: 10 ckpt_save_dir: './ckpt' epoch_size: 200 dataset_sink_mode: True -amp_level: 'O2' +amp_level: 'O0' # loss loss: 'CE' From 50f9abae8324e5200e763606946896846176996d Mon Sep 17 00:00:00 2001 From: PingqiLi Date: Wed, 22 May 2024 15:25:13 +0800 Subject: [PATCH 2/2] Modify the ssd/data.py to make it compatible with the mindspore v2.0+ version. --- examples/det/ssd/data.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/examples/det/ssd/data.py b/examples/det/ssd/data.py index 31d82ca0..1aa06d22 100644 --- a/examples/det/ssd/data.py +++ b/examples/det/ssd/data.py @@ -1,3 +1,4 @@ +import inspect import os import cv2 @@ -175,14 +176,21 @@ def compose_map_func(img_id, image, annotation): output_columns = ["img_id", "image", "image_shape"] trans = [normalize_op, change_swap_op] + # Note: mindspore-2.0 delete the parameter column_order + sig = inspect.signature(ds.map) + pass_column_order = False if "kwargs" in sig.parameters else True + ds = ds.map( operations=compose_map_func, input_columns=["img_id", "image", "annotation"], output_columns=output_columns, - column_order=output_columns, + column_order=output_columns if pass_column_order else None, python_multiprocessing=python_multiprocessing, num_parallel_workers=num_parallel_workers, ) + if not pass_column_order: + ds.project(output_columns=output_columns) + ds = ds.map( operations=trans, input_columns=["image"],