diff --git a/configs/shufflenet_v1/README.md b/configs/shufflenet_v1/README.md index 7866d4a1..a7dd1d53 100644 --- a/configs/shufflenet_v1/README.md +++ b/configs/shufflenet_v1/README.md @@ -24,8 +24,6 @@ Our reproduced model performance on ImageNet-1K is reported as follows. |-----------------------|----------|-----------|-----------|------------|---------------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------| | shufflenet_v1_g3_x0_5 | D910x8-G | 57.05 | 79.73 | 0.73 | [yaml](https://github.com/mindspore-lab/mindcv/blob/main/configs/shufflenet_v1/shufflenet_v1_0.5_ascend.yaml) | [weights](https://download.mindspore.cn/toolkits/mindcv/shufflenet/shufflenetv1/shufflenet_v1_g3_05-42cfe109.ckpt) | | shufflenet_v1_g3_x1_0 | D910x8-G | 67.77 | 87.73 | 1.89 | [yaml](https://github.com/mindspore-lab/mindcv/blob/main/configs/shufflenet_v1/shufflenet_v1_1.0_ascend.yaml) | [weights](https://download.mindspore.cn/toolkits/mindcv/shufflenet/shufflenetv1/shufflenet_v1_g3_10-245f0ccf.ckpt) | -| shufflenet_v1_g3_x1_5 | D910x8-G | 71.53 | 90.17 | 3.48 | [yaml](https://github.com/mindspore-lab/mindcv/blob/main/configs/shufflenet_v1/shufflenet_v1_1.5_ascend.yaml) | [weights](https://download.mindspore.cn/toolkits/mindcv/shufflenet/shufflenetv1/shufflenet_v1_g3_15-40caaa99.ckpt) | -| shufflenet_v1_g3_x2_0 | D910x8-G | 74.02 | 91.74 | 5.50 | [yaml](https://github.com/mindspore-lab/mindcv/blob/main/configs/shufflenet_v1/shufflenet_v1_2.0_ascend.yaml) | [weights](https://download.mindspore.cn/toolkits/mindcv/shufflenet/shufflenetv1/shufflenet_v1_g3_20-df3f5199.ckpt) | diff --git a/configs/shufflenet_v1/shufflenet_v1_1.5_ascend.yaml b/configs/shufflenet_v1/shufflenet_v1_1.5_ascend.yaml deleted file mode 100644 index 876b838b..00000000 --- a/configs/shufflenet_v1/shufflenet_v1_1.5_ascend.yaml +++ /dev/null @@ -1,49 +0,0 @@ -# system -mode: 0 -distribute: True -num_parallel_workers: 8 - -# dataset -dataset: "imagenet" -data_dir: "/path/to/imagenet" -shuffle: True -dataset_download: False -batch_size: 64 -drop_remainder: True - -# augmentation -image_resize: 224 -hflip: 0.5 -color_jitter: 0.4 -interpolation: "bilinear" -crop_pct: 0.875 - -# model -model: "shufflenet_v1_g3_x1_5" -num_classes: 1000 -pretrained: False -ckpt_path: "" -keep_checkpoint_max: 30 -ckpt_save_dir: "./ckpt" -epoch_size: 250 -dataset_sink_mode: True -amp_level: "O0" - -# loss -loss: "CE" -label_smoothing: 0.1 - -# lr scheduler -scheduler: "cosine_decay" -min_lr: 0.0 -lr: 0.5 -warmup_epochs: 4 -decay_epochs: 246 - -# optimizer -opt: "momentum" -filter_bias_and_bn: False -momentum: 0.9 -weight_decay: 0.00004 -loss_scale: 1024 -use_nesterov: False diff --git a/configs/shufflenet_v1/shufflenet_v1_2.0_ascend.yaml b/configs/shufflenet_v1/shufflenet_v1_2.0_ascend.yaml deleted file mode 100644 index f0faf4e0..00000000 --- a/configs/shufflenet_v1/shufflenet_v1_2.0_ascend.yaml +++ /dev/null @@ -1,49 +0,0 @@ -# system -mode: 0 -distribute: True -num_parallel_workers: 8 - -# dataset -dataset: "imagenet" -data_dir: "/path/to/imagenet" -shuffle: True -dataset_download: False -batch_size: 64 -drop_remainder: True - -# augmentation -image_resize: 224 -hflip: 0.5 -color_jitter: 0.4 -interpolation: "bilinear" -crop_pct: 0.875 - -# model -model: "shufflenet_v1_g3_x2_0" -num_classes: 1000 -pretrained: False -ckpt_path: "" -keep_checkpoint_max: 30 -ckpt_save_dir: "./ckpt" -epoch_size: 250 -dataset_sink_mode: True -amp_level: "O0" - -# loss -loss: "CE" -label_smoothing: 0.1 - -# lr scheduler -scheduler: "cosine_decay" -min_lr: 0.0 -lr: 0.5 -warmup_epochs: 4 -decay_epochs: 246 - -# optimizer -opt: "momentum" -filter_bias_and_bn: False -momentum: 0.9 -weight_decay: 0.00004 -loss_scale: 1024 -use_nesterov: False diff --git a/mindcv/models/shufflenetv1.py b/mindcv/models/shufflenetv1.py index bffd6c43..2293a250 100644 --- a/mindcv/models/shufflenetv1.py +++ b/mindcv/models/shufflenetv1.py @@ -40,12 +40,8 @@ def _cfg(url="", **kwargs): "shufflenet_v1_g3_1.0": _cfg( url="https://download.mindspore.cn/toolkits/mindcv/shufflenet/shufflenetv1/shufflenet_v1_g3_10-245f0ccf.ckpt" ), - "shufflenet_v1_g3_1.5": _cfg( - url="https://download.mindspore.cn/toolkits/mindcv/shufflenet/shufflenetv1/shufflenet_v1_g3_15-40caaa99.ckpt" - ), - "shufflenet_v1_g3_2.0": _cfg( - url="https://download.mindspore.cn/toolkits/mindcv/shufflenet/shufflenetv1/shufflenet_v1_g3_20-df3f5199.ckpt" - ), + "shufflenet_v1_g3_1.5": _cfg(url=""), + "shufflenet_v1_g3_2.0": _cfg(url=""), "shufflenet_v1_g8_0.5": _cfg(url=""), "shufflenet_v1_g8_1.0": _cfg(url=""), "shufflenet_v1_g8_1.5": _cfg(url=""),