#Configuration


1.5. Introduction

This document introduces the configuration(filed in config/*.yaml) of PaddleClas.

1.5.1. Basic

name detail default value optional value
mode mode "train" ["train"," valid"]
architecture model name "ResNet50_vd" one of 23 architectures
pretrained_model pretrained model path "" Str
model_save_dir model stored path "" Str
classes_num class number 1000 int
total_images total images 1281167 int
save_interval save interval 1 int
validate whether to validate when training TRUE bool
valid_interval valid interval 1 int
epochs epoch int
topk K value 5 int
image_shape image size [3,224,224] list, shape: (3,)
use_mix whether to use mixup False ['True', 'False']
ls_epsilon label_smoothing epsilon value 0 float

1.5.2. Optimizer & Learning rate

learning rate

name detail default value Optional value
function decay type "Linear" ["Linear", "Cosine",
"Piecewise", "CosineWarmup"]
params.lr initial learning rate 0.1 float
params.decay_epochs milestone in piecewisedecay list
params.gamma gamma in piecewisedecay 0.1 float
params.warmup_epoch warmup epoch 5 int
parmas.steps decay steps in lineardecay 100 int
params.end_lr end lr in lineardecay 0 float

optimizer

name detail default value optional value
function optimizer name "Momentum" ["Momentum", "RmsProp"]
params.momentum momentum value 0.9 float
regularizer.function regularizer method name "L2" ["L1", "L2"]
regularizer.factor regularizer factor 0.0001 float

1.5.3. reader

name detail
batch_size batch size
num_workers worker number
file_list train list path
data_dir train dataset path
shuffle_seed seed

processing

function name attribute name detail
DecodeImage to_rgb decode to RGB
to_np to numpy
channel_first Channel first
RandCropImage size random crop
RandFlipImage random flip
NormalizeImage scale normalize image
mean mean
std std
order order
ToCHWImage to CHW
CropImage size crop size
ResizeImage resize_short resize according to short size

mix preprocessing

name detail
MixupOperator.alpha alpha value in mixup