From 083a77e763de1f578c2ce2d0901a41447f201cfb Mon Sep 17 00:00:00 2001 From: weishengyu Date: Fri, 2 Oct 2020 03:28:51 +0800 Subject: [PATCH] add config --- configs/GhostNet/GhostNet_x0_5.yaml | 74 +++++++++++++++++++++++ configs/GhostNet/GhostNet_x1_0.yaml | 74 +++++++++++++++++++++++ configs/GhostNet/GhostNet_x1_3.yaml | 75 ++++++++++++++++++++++++ ppcls/modeling/architectures/ghostnet.py | 6 +- 4 files changed, 226 insertions(+), 3 deletions(-) create mode 100644 configs/GhostNet/GhostNet_x0_5.yaml create mode 100644 configs/GhostNet/GhostNet_x1_0.yaml create mode 100644 configs/GhostNet/GhostNet_x1_3.yaml diff --git a/configs/GhostNet/GhostNet_x0_5.yaml b/configs/GhostNet/GhostNet_x0_5.yaml new file mode 100644 index 00000000..e787c6bd --- /dev/null +++ b/configs/GhostNet/GhostNet_x0_5.yaml @@ -0,0 +1,74 @@ +mode: 'train' +ARCHITECTURE: + name: 'GhostNet_x0_5' + +pretrained_model: "" +model_save_dir: "./output/" +classes_num: 1000 +total_images: 1281167 +save_interval: 1 +validate: True +valid_interval: 1 +epochs: 360 +topk: 5 +image_shape: [3, 224, 224] + +use_mix: False +ls_epsilon: 0.1 + +LEARNING_RATE: + function: 'CosineWarmup' + params: + lr: 0.8 + +OPTIMIZER: + function: 'Momentum' + params: + momentum: 0.9 + regularizer: + function: 'L2' + factor: 0.0000400 + +TRAIN: + batch_size: 2048 + num_workers: 4 + file_list: "./dataset/ILSVRC2012/train_list.txt" + data_dir: "./dataset/ILSVRC2012/" + shuffle_seed: 0 + transforms: + - DecodeImage: + to_rgb: True + to_np: False + channel_first: False + - RandCropImage: + size: 224 + - RandFlipImage: + flip_code: 1 + - NormalizeImage: + scale: 1./255. + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + - ToCHWImage: + +VALID: + batch_size: 64 + num_workers: 4 + file_list: "./dataset/ILSVRC2012/val_list.txt" + data_dir: "./dataset/ILSVRC2012/" + shuffle_seed: 0 + transforms: + - DecodeImage: + to_rgb: True + to_np: False + 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: diff --git a/configs/GhostNet/GhostNet_x1_0.yaml b/configs/GhostNet/GhostNet_x1_0.yaml new file mode 100644 index 00000000..c1a61dca --- /dev/null +++ b/configs/GhostNet/GhostNet_x1_0.yaml @@ -0,0 +1,74 @@ +mode: 'train' +ARCHITECTURE: + name: 'GhostNet_x1_0' + +pretrained_model: "" +model_save_dir: "./output/" +classes_num: 1000 +total_images: 1281167 +save_interval: 1 +validate: True +valid_interval: 1 +epochs: 360 +topk: 5 +image_shape: [3, 224, 224] + +use_mix: False +ls_epsilon: 0.1 + +LEARNING_RATE: + function: 'CosineWarmup' + params: + lr: 0.4 + +OPTIMIZER: + function: 'Momentum' + params: + momentum: 0.9 + regularizer: + function: 'L2' + factor: 0.0000400 + +TRAIN: + batch_size: 1024 + num_workers: 4 + file_list: "./dataset/ILSVRC2012/train_list.txt" + data_dir: "./dataset/ILSVRC2012/" + shuffle_seed: 0 + transforms: + - DecodeImage: + to_rgb: True + to_np: False + channel_first: False + - RandCropImage: + size: 224 + - RandFlipImage: + flip_code: 1 + - NormalizeImage: + scale: 1./255. + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + - ToCHWImage: + +VALID: + batch_size: 64 + num_workers: 4 + file_list: "./dataset/ILSVRC2012/val_list.txt" + data_dir: "./dataset/ILSVRC2012/" + shuffle_seed: 0 + transforms: + - DecodeImage: + to_rgb: True + to_np: False + 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: diff --git a/configs/GhostNet/GhostNet_x1_3.yaml b/configs/GhostNet/GhostNet_x1_3.yaml new file mode 100644 index 00000000..452c1f49 --- /dev/null +++ b/configs/GhostNet/GhostNet_x1_3.yaml @@ -0,0 +1,75 @@ +mode: 'train' +ARCHITECTURE: + name: 'GhostNet_x1_3' + +pretrained_model: "" +model_save_dir: "./output/" +classes_num: 1000 +total_images: 1281167 +save_interval: 1 +validate: True +valid_interval: 1 +epochs: 360 +topk: 5 +image_shape: [3, 224, 224] + +use_mix: False +ls_epsilon: 0.1 + +LEARNING_RATE: + function: 'CosineWarmup' + params: + lr: 0.4 + +OPTIMIZER: + function: 'Momentum' + params: + momentum: 0.9 + regularizer: + function: 'L2' + factor: 0.0000400 + +TRAIN: + batch_size: 1024 + num_workers: 4 + file_list: "./dataset/ILSVRC2012/train_list.txt" + data_dir: "./dataset/ILSVRC2012/" + shuffle_seed: 0 + transforms: + - DecodeImage: + to_rgb: True + to_np: False + channel_first: False + - RandCropImage: + size: 224 + - RandFlipImage: + flip_code: 1 + - AutoAugment: + - NormalizeImage: + scale: 1./255. + mean: [0.485, 0.456, 0.406] + std: [0.229, 0.224, 0.225] + order: '' + - ToCHWImage: + +VALID: + batch_size: 64 + num_workers: 4 + file_list: "./dataset/ILSVRC2012/val_list.txt" + data_dir: "./dataset/ILSVRC2012/" + shuffle_seed: 0 + transforms: + - DecodeImage: + to_rgb: True + to_np: False + 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: diff --git a/ppcls/modeling/architectures/ghostnet.py b/ppcls/modeling/architectures/ghostnet.py index 3c8dafbc..bea215be 100644 --- a/ppcls/modeling/architectures/ghostnet.py +++ b/ppcls/modeling/architectures/ghostnet.py @@ -355,16 +355,16 @@ class GhostNet(nn.Layer): return new_v -def GhostNet_x0_5(): +def GhostNet_x0_5(**args): model = GhostNet(scale=0.5) return model -def GhostNet_x1_0(): +def GhostNet_x1_0(**args): model = GhostNet(scale=1.0) return model -def GhostNet_x1_3(): +def GhostNet_x1_3(**args): model = GhostNet(scale=1.3) return model \ No newline at end of file -- GitLab