From b1a51f6ad560548bd061a440e11dc990b5617895 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Wed, 18 Aug 2021 05:08:40 +0000 Subject: [PATCH] Update mobile model configs --- .../ImageNet/MobileNetV1/MobileNetV1.yaml | 2 +- .../MobileNetV1/MobileNetV1_x0_25.yaml | 2 +- .../MobileNetV1/MobileNetV1_x0_5.yaml | 2 +- .../MobileNetV1/MobileNetV1_x0_75.yaml | 2 +- .../ImageNet/MobileNetV2/MobileNetV2.yaml | 2 +- .../MobileNetV2/MobileNetV2_x0_25.yaml | 2 +- .../MobileNetV2/MobileNetV2_x0_5.yaml | 2 +- .../MobileNetV2/MobileNetV2_x0_75.yaml | 2 +- .../MobileNetV2/MobileNetV2_x1_5.yaml | 2 +- .../MobileNetV2/MobileNetV2_x2_0.yaml | 2 +- .../ShuffleNet/ShuffleNetV2_swish.yaml | 129 ++++++++++++++++++ .../ShuffleNet/ShuffleNetV2_x0_25.yaml | 2 +- .../ShuffleNet/ShuffleNetV2_x0_33.yaml | 2 +- .../ShuffleNet/ShuffleNetV2_x0_5.yaml | 2 +- .../ShuffleNet/ShuffleNetV2_x1_0.yaml | 2 +- .../ShuffleNet/ShuffleNetV2_x1_5.yaml | 2 +- .../ShuffleNet/ShuffleNetV2_x2_0.yaml | 2 +- 17 files changed, 145 insertions(+), 16 deletions(-) create mode 100644 ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_swish.yaml diff --git a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml index a39496d7..53d1d1dd 100644 --- a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml +++ b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml @@ -41,7 +41,7 @@ Optimizer: values: [0.1, 0.01, 0.001, 0.0001] regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_25.yaml b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_25.yaml index b4de5a91..9fc0dd98 100644 --- a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_25.yaml +++ b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_25.yaml @@ -39,7 +39,7 @@ Optimizer: values: [0.1, 0.01, 0.001, 0.0001] regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_5.yaml b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_5.yaml index 1728d0a7..ef709640 100644 --- a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_5.yaml +++ b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_5.yaml @@ -39,7 +39,7 @@ Optimizer: values: [0.1, 0.01, 0.001, 0.0001] regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_75.yaml b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_75.yaml index ca80fb0b..b9be283f 100644 --- a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_75.yaml +++ b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1_x0_75.yaml @@ -39,7 +39,7 @@ Optimizer: values: [0.1, 0.01, 0.001, 0.0001] regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml index 40fc2d1f..fc0e61bb 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml @@ -39,7 +39,7 @@ Optimizer: learning_rate: 0.045 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_25.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_25.yaml index a08cee87..0ebe4589 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_25.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_25.yaml @@ -37,7 +37,7 @@ Optimizer: learning_rate: 0.045 regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_5.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_5.yaml index f541d244..c2023908 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_5.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_5.yaml @@ -37,7 +37,7 @@ Optimizer: learning_rate: 0.045 regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_75.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_75.yaml index 1ade41b7..e581d72f 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_75.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x0_75.yaml @@ -37,7 +37,7 @@ Optimizer: learning_rate: 0.045 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x1_5.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x1_5.yaml index 555f6a20..2ac0d699 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x1_5.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x1_5.yaml @@ -37,7 +37,7 @@ Optimizer: learning_rate: 0.045 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x2_0.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x2_0.yaml index 03fba8d3..ca4bf665 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x2_0.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2_x2_0.yaml @@ -37,7 +37,7 @@ Optimizer: learning_rate: 0.045 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_swish.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_swish.yaml new file mode 100644 index 00000000..79bb34ed --- /dev/null +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_swish.yaml @@ -0,0 +1,129 @@ +# global configs +Global: + checkpoints: null + pretrained_model: null + output_dir: ./output/ + device: gpu + save_interval: 1 + eval_during_train: True + eval_interval: 1 + epochs: 240 + print_batch_step: 10 + use_visualdl: False + # used for static mode and model export + image_shape: [3, 224, 224] + save_inference_dir: ./inference + +# model architecture +Arch: + name: ShuffleNetV2_swish + class_num: 1000 + +# loss function config for traing/eval process +Loss: + Train: + - CELoss: + weight: 1.0 + Eval: + - CELoss: + weight: 1.0 + + +Optimizer: + name: Momentum + momentum: 0.9 + lr: + name: Cosine + learning_rate: 0.5 + warmup_epoch: 5 + regularizer: + name: 'L2' + coeff: 0.00004 + + +# data loader for train and eval +DataLoader: + Train: + dataset: + name: ImageNetDataset + image_root: ./dataset/ILSVRC2012/ + cls_label_path: ./dataset/ILSVRC2012/train_list.txt + transform_ops: + - DecodeImage: + to_rgb: True + channel_first: False + - RandCropImage: + size: 224 + - RandFlipImage: + flip_code: 1 + - NormalizeImage: + scale: 1.0/255.0 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + + sampler: + name: DistributedBatchSampler + batch_size: 256 + drop_last: False + shuffle: True + loader: + num_workers: 4 + use_shared_memory: True + + Eval: + dataset: + name: ImageNetDataset + image_root: ./dataset/ILSVRC2012/ + cls_label_path: ./dataset/ILSVRC2012/val_list.txt + transform_ops: + - DecodeImage: + to_rgb: True + channel_first: False + - ResizeImage: + resize_short: 256 + - CropImage: + size: 224 + - NormalizeImage: + scale: 1.0/255.0 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + sampler: + name: DistributedBatchSampler + batch_size: 64 + drop_last: False + shuffle: False + loader: + num_workers: 4 + use_shared_memory: True + +Infer: + infer_imgs: docs/images/whl/demo.jpg + batch_size: 10 + transforms: + - DecodeImage: + to_rgb: True + channel_first: False + - ResizeImage: + resize_short: 256 + - CropImage: + size: 224 + - NormalizeImage: + scale: 1.0/255.0 + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + - ToCHWImage: + PostProcess: + name: Topk + topk: 5 + class_id_map_file: ppcls/utils/imagenet1k_label_list.txt + +Metric: + Train: + - TopkAcc: + topk: [1, 5] + Eval: + - TopkAcc: + topk: [1, 5] diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_25.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_25.yaml index ec2ecdbd..aa1cce18 100644 --- a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_25.yaml +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_25.yaml @@ -38,7 +38,7 @@ Optimizer: warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_33.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_33.yaml index cf9781a1..3f0742e5 100644 --- a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_33.yaml +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_33.yaml @@ -38,7 +38,7 @@ Optimizer: warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_5.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_5.yaml index dc7376af..f14a2492 100644 --- a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_5.yaml +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x0_5.yaml @@ -38,7 +38,7 @@ Optimizer: warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0003 + coeff: 0.00003 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml index 8a7d4195..dd4820da 100644 --- a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_0.yaml @@ -38,7 +38,7 @@ Optimizer: warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_5.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_5.yaml index a8cc24fa..9a05a59d 100644 --- a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_5.yaml +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x1_5.yaml @@ -38,7 +38,7 @@ Optimizer: warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x2_0.yaml b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x2_0.yaml index d92132a8..c871ec7d 100644 --- a/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x2_0.yaml +++ b/ppcls/configs/ImageNet/ShuffleNet/ShuffleNetV2_x2_0.yaml @@ -38,7 +38,7 @@ Optimizer: warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval -- GitLab