2020-06-05 16:55:47 +08:00
|
|
|
|
#Configuration
|
|
|
|
|
|
|
|
|
|
|
|
---
|
|
|
|
|
|
|
|
|
|
|
|
## Introduction
|
|
|
|
|
|
|
2020-06-18 19:45:11 +08:00
|
|
|
|
This document introduces the configuration(filed in `config/*.yaml`) of PaddleClas.
|
2020-06-05 16:55:47 +08:00
|
|
|
|
|
|
|
|
|
|
### 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", <br> "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|
|
