2.8 KiB

Raw Blame History

配置说明

简介

本文档介绍了PaddleClas配置文件(configs/*.yaml)中各参数的含义，以便您更快的自定义或修改超参数配置。

配置详解

基础配置

参数名字	具体含义	默认值	可选值
mode	运行模式	"train"	["train"," valid"]
architecture	模型结构名字	"ResNet50_vd"	PaddleClas提供的模型结构
pretrained_model	预训练模型路径	""	Str
model_save_dir	保存模型路径	""	Str
classes_num	分类数	1000	int
total_images	总图片数	1281167	int
save_interval	每隔多少个epoch保存模型	1	int
validate	是否在训练时进行评估	TRUE	bool
valid_interval	每隔多少个epoch进行模型评估	1	int
epochs	训练总epoch数		int
topk	评估指标K值大小	5	int
image_shape	图片大小	[3，224，224]	list, shape: (3,)
use_mix	是否启用mixup	False	['True', 'False']
ls_epsilon	label_smoothing epsilon值	0	float

学习率与优化器

学习率

参数名字	具体含义	默认值	可选值
function	decay方法名	"Linear"	["Linear", "Cosine", "Piecewise", "CosineWarmup"]
params.lr	初始学习率	0.1	float
params.decay_epochs	piecewisedecay中衰减学习率的milestone		list
params.gamma	piecewisedecay中gamma值	0.1	float
params.warmup_epoch	warmup轮数	5	int
parmas.steps	lineardecay衰减steps数	100	int
params.end_lr	lineardecayend_lr值	0	float

优化器

参数名字	具体含义	默认值	可选值
function	优化器方法名	"Momentum"	["Momentum", "RmsProp"]
params.momentum	momentum值	0.9	float
regularizer.function	正则化方法名	"L2"	["L1", "L2"]
regularizer.factor	正则化系数	0.0001	float

数据读取器与数据处理

参数名字	具体含义
batch_size	批大小
num_workers	数据读取器worker数量
file_list	train文件列表
data_dir	train文件路径
shuffle_seed	用来进行shuffle的seed值

数据处理

功能名字	参数名字	具体含义
DecodeImage	to_rgb	数据转RGB
	to_np	数据转numpy
	channel_first	按CHW排列的图片数据
RandCropImage	size	随机裁剪
RandFlipImage		随机翻转
NormalizeImage	scale	归一化scale值
	mean	归一化均值
	std	归一化方差
	order	归一化顺序
ToCHWImage		调整为CHW
CropImage	size	裁剪大小
ResizeImage	resize_short	按短边调整大小

mix处理

参数名字	具体含义
MixupOperator.alpha	mixup处理中的alpha值