| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- # global configs
- Global:
- checkpoints: null
- pretrained_model: null
- output_dir: ./output/
- device: gpu
- save_interval: 1
- eval_during_train: True
- eval_interval: 1
- epochs: 600
- print_batch_step: 10
- use_visualdl: False
- # used for static mode and model export
- image_shape: [3, 224, 224]
- save_inference_dir: ./inference
- # training model under @to_static
- to_static: False
- use_dali: False
- # mixed precision training
- AMP:
- use_amp: True
- use_fp16_test: False
- scale_loss: 128.0
- use_dynamic_loss_scaling: True
- use_promote: False
- # O1: mixed fp16, O2: pure fp16
- level: O1
- # model architecture
- Arch:
- name: PPHGNet_base
- class_num: 102
-
- # loss function config for traing/eval process
- Loss:
- Train:
- - CELoss:
- weight: 1.0
- epsilon: 0.1
- Eval:
- - CELoss:
- weight: 1.0
- Optimizer:
- name: Momentum
- momentum: 0.9
- lr:
- name: Cosine
- learning_rate: 0.5
- warmup_epoch: 5
- regularizer:
- name: 'L2'
- coeff: 0.00004
- # data loader for train and eval
- DataLoader:
- Train:
- dataset:
- name: ImageNetDataset
- image_root: ./dataset/ILSVRC2012/
- cls_label_path: ./dataset/ILSVRC2012/train_list.txt
- transform_ops:
- - DecodeImage:
- to_rgb: True
- channel_first: False
- - RandCropImage:
- size: 224
- interpolation: bicubic
- backend: pil
- - RandFlipImage:
- flip_code: 1
- - TimmAutoAugment:
- config_str: rand-m15-mstd0.5-inc1
- interpolation: bicubic
- img_size: 224
- - NormalizeImage:
- scale: 1.0/255.0
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- order: ''
- - RandomErasing:
- EPSILON: 0.4
- sl: 0.02
- sh: 1.0/3.0
- r1: 0.3
- attempt: 10
- use_log_aspect: True
- mode: pixel
- batch_transform_ops:
- - OpSampler:
- MixupOperator:
- alpha: 0.4
- prob: 0.5
- CutmixOperator:
- alpha: 1.0
- prob: 0.5
- sampler:
- name: DistributedBatchSampler
- batch_size: 128
- drop_last: False
- shuffle: True
- loader:
- num_workers: 16
- use_shared_memory: True
- Eval:
- dataset:
- name: ImageNetDataset
- image_root: ./dataset/ILSVRC2012/
- cls_label_path: ./dataset/ILSVRC2012/val_list.txt
- transform_ops:
- - DecodeImage:
- to_rgb: True
- channel_first: False
- - ResizeImage:
- resize_short: 236
- interpolation: bicubic
- backend: pil
- - CropImage:
- size: 224
- - NormalizeImage:
- scale: 1.0/255.0
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- order: ''
- sampler:
- name: DistributedBatchSampler
- batch_size: 128
- drop_last: False
- shuffle: False
- loader:
- num_workers: 16
- use_shared_memory: True
- Infer:
- infer_imgs: docs/images/inference_deployment/whl_demo.jpg
- batch_size: 10
- transforms:
- - DecodeImage:
- to_rgb: True
- channel_first: False
- - ResizeImage:
- resize_short: 236
- interpolation: bicubic
- backend: pil
- - CropImage:
- size: 224
- - NormalizeImage:
- scale: 1.0/255.0
- mean: [0.485, 0.456, 0.406]
- std: [0.229, 0.224, 0.225]
- order: ''
- - ToCHWImage:
- PostProcess:
- name: Topk
- topk: 5
- class_id_map_file: ppcls/utils/imagenet1k_label_list.txt
- Metric:
- Train:
- - TopkAcc:
- topk: [1, 5]
- Eval:
- - TopkAcc:
- topk: [1, 5]
|
