配置说明¶
简介¶
本文档介绍了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值 |