diff --git a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml index a39496d7b37fe0143c56cfce88ebe7dba5e48989..53d1d1dd27659da5be30f4e9b21ce8084c332f57 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 b4de5a9131c94acc6f671b75bcaa52d605fcc7ac..9fc0dd98910df880d95c5ef030ec7eae16deac5f 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 1728d0a722b4cb782e9fe11ae3b6d82f4ac05684..ef70964095432d5dc97ce97835064599d85dca6a 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 ca80fb0b7213f558a0f959697231b6627ab42561..b9be283f39e37e32428c5b316db290fd57794a0d 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 40fc2d1f4fb81a33ed6b66f29b6cf6d9bbdae6a3..fc0e61bb6006e48d6b5856b2db5c1ff286cdd70e 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 a08cee8798daaea22ca1fe8b687287fc9e4d1365..0ebe45893a7fee0de1678b37121cf7ec8c2d9df0 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 f541d24477b0f1cd3a05d4e88e2e7db6548cd9b5..c2023908f13b40ed8580703a6deb9ac6d9f59c82 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 1ade41b7e830a0a3deefc6b23001137d25b02219..e581d72f9d9da3ac966ddbfa793d014b721277fd 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 555f6a207dd77acf52a26e1d50688780850b6c82..2ac0d6992df480166b52b277116a8c98ab376fca 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 03fba8d32a28ceeea1800a743d21a97460f11429..ca4bf665039e8eaa5c6f54cf44dd71486b0df440 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 0000000000000000000000000000000000000000..79bb34edec97c1af7398cbd36afa3bc1ed6b5bc4 --- /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 ec2ecdbd46663fcafe7148d6fc9842459f92ff28..aa1cce189d4d9bc58166d42466f997750cbfc8c6 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 cf9781a16c9fe9b3a2c529d4b5ce0617bc1d0057..3f0742e542eac487930b7713b5885a2fb75763c1 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 dc7376af89ae166483a2c8033d8d7193296195ab..f14a24923998bc635e1e25b4a72475d4f893ea1b 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 8a7d41952b4dedec8f6376b791b8d8596e628f58..dd4820da45ab49302510ed834bce584613721e05 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 a8cc24fa69b4ac5d76531be21264edcf70302009..9a05a59d60b6814299d338cf638be14f29ff5873 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 d92132a8f5f6c852efa770bf9345cb3884129da8..c871ec7dcc7aa84158b9c20ab75057f499d8d0bb 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