PaddleClas/docs/en/tutorials/config_en.md

#Configuration

---

Introduction

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

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

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

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