mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
add cspresnet50 config
This commit is contained in:
parent
07e1e5dc31
commit
88e671b2f3
76
configs/CSPNet/CSPResNet50.yaml
Normal file
76
configs/CSPNet/CSPResNet50.yaml
Normal file
@ -0,0 +1,76 @@
|
|||||||
|
mode: 'train'
|
||||||
|
ARCHITECTURE:
|
||||||
|
name: 'CSPResNet50'
|
||||||
|
|
||||||
|
pretrained_model: ""
|
||||||
|
model_save_dir: "./output/"
|
||||||
|
classes_num: 1000
|
||||||
|
total_images: 1281167
|
||||||
|
save_interval: 1
|
||||||
|
validate: True
|
||||||
|
valid_interval: 1
|
||||||
|
epochs: 120
|
||||||
|
topk: 5
|
||||||
|
image_shape: [3, 256, 256]
|
||||||
|
|
||||||
|
use_mix: False
|
||||||
|
ls_epsilon: -1
|
||||||
|
|
||||||
|
LEARNING_RATE:
|
||||||
|
function: 'Piecewise'
|
||||||
|
params:
|
||||||
|
lr: 0.1
|
||||||
|
decay_epochs: [30, 60, 90]
|
||||||
|
gamma: 0.1
|
||||||
|
|
||||||
|
OPTIMIZER:
|
||||||
|
function: 'Momentum'
|
||||||
|
params:
|
||||||
|
momentum: 0.9
|
||||||
|
regularizer:
|
||||||
|
function: 'L2'
|
||||||
|
factor: 0.000100
|
||||||
|
|
||||||
|
TRAIN:
|
||||||
|
batch_size: 256
|
||||||
|
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: 256
|
||||||
|
- 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: 256
|
||||||
|
- NormalizeImage:
|
||||||
|
scale: 1.0/255.0
|
||||||
|
mean: [0.485, 0.456, 0.406]
|
||||||
|
std: [0.229, 0.224, 0.225]
|
||||||
|
order: ''
|
||||||
|
- ToCHWImage:
|
Loading…
x
Reference in New Issue
Block a user