MobileNetV2.yaml 2.7 KB
Newer Older
1 2 3 4
# global configs
Global:
  checkpoints: null
  pretrained_model: null
C
cuicheng01 已提交
5 6
  output_dir: ./output/
  device: gpu
7 8 9
  save_interval: 1
  eval_during_train: True
  eval_interval: 1
10
  epochs: 240
11 12 13
  print_batch_step: 10
  use_visualdl: False
  # used for static mode and model export
14
  image_shape: [3, 224, 224]
C
cuicheng01 已提交
15
  save_inference_dir: ./inference
A
Aurelius84 已提交
16 17
  # training model under @to_static
  to_static: False
18 19 20

# model architecture
Arch:
21
  name: MobileNetV2
littletomatodonkey's avatar
littletomatodonkey 已提交
22
  class_num: 1000
23 24 25 26 27 28 29 30 31 32 33 34
 
# loss function config for traing/eval process
Loss:
  Train:
    - CELoss:
        weight: 1.0
  Eval:
    - CELoss:
        weight: 1.0


Optimizer:
C
cuicheng01 已提交
35
  name: Momentum
36 37
  momentum: 0.9
  lr:
C
cuicheng01 已提交
38
    name: Cosine
39 40 41
    learning_rate: 0.045
  regularizer:
    name: 'L2'
C
cuicheng01 已提交
42
    coeff: 0.00004
43 44 45 46 47 48


# data loader for train and eval
DataLoader:
  Train:
    dataset:
C
cuicheng01 已提交
49 50 51
      name: ImageNetDataset
      image_root: ./dataset/ILSVRC2012/
      cls_label_path: ./dataset/ILSVRC2012/train_list.txt
C
cuicheng01 已提交
52
      transform_ops:
53 54 55
        - DecodeImage:
            to_rgb: True
            channel_first: False
C
cuicheng01 已提交
56
        - RandCropImage:
57
            size: 224
C
cuicheng01 已提交
58 59 60 61 62 63 64
        - RandFlipImage:
            flip_code: 1
        - NormalizeImage:
            scale: 1.0/255.0
            mean: [0.485, 0.456, 0.406]
            std: [0.229, 0.224, 0.225]
            order: ''
65 66

    sampler:
C
cuicheng01 已提交
67
      name: DistributedBatchSampler
C
cuicheng01 已提交
68 69 70
      batch_size: 64
      drop_last: False
      shuffle: True
71
    loader:
C
cuicheng01 已提交
72 73
      num_workers: 4
      use_shared_memory: True
74 75 76

  Eval:
    dataset: 
C
cuicheng01 已提交
77 78 79
      name: ImageNetDataset
      image_root: ./dataset/ILSVRC2012/
      cls_label_path: ./dataset/ILSVRC2012/val_list.txt
C
cuicheng01 已提交
80
      transform_ops:
81 82 83
        - DecodeImage:
            to_rgb: True
            channel_first: False
C
cuicheng01 已提交
84
        - ResizeImage:
85
            resize_short: 256
C
cuicheng01 已提交
86
        - CropImage:
87
            size: 224
C
cuicheng01 已提交
88 89 90 91 92
        - NormalizeImage:
            scale: 1.0/255.0
            mean: [0.485, 0.456, 0.406]
            std: [0.229, 0.224, 0.225]
            order: ''
93
    sampler:
C
cuicheng01 已提交
94
      name: DistributedBatchSampler
C
cuicheng01 已提交
95 96 97
      batch_size: 64
      drop_last: False
      shuffle: False
98
    loader:
C
cuicheng01 已提交
99 100
      num_workers: 4
      use_shared_memory: True
101 102

Infer:
T
Tingquan Gao 已提交
103
  infer_imgs: docs/images/inference_deployment/whl_demo.jpg
104 105
  batch_size: 10
  transforms:
C
cuicheng01 已提交
106 107 108 109
    - DecodeImage:
        to_rgb: True
        channel_first: False
    - ResizeImage:
110
        resize_short: 256
C
cuicheng01 已提交
111
    - CropImage:
112
        size: 224
C
cuicheng01 已提交
113 114 115 116 117 118
    - NormalizeImage:
        scale: 1.0/255.0
        mean: [0.485, 0.456, 0.406]
        std: [0.229, 0.224, 0.225]
        order: ''
    - ToCHWImage:
119
  PostProcess:
C
cuicheng01 已提交
120
    name: Topk
121
    topk: 5
C
cuicheng01 已提交
122
    class_id_map_file: ppcls/utils/imagenet1k_label_list.txt
123 124

Metric:
C
cuicheng01 已提交
125
  Train:
C
cuicheng01 已提交
126 127
    - TopkAcc:
        topk: [1, 5]
C
cuicheng01 已提交
128
  Eval:
C
cuicheng01 已提交
129 130
    - TopkAcc:
        topk: [1, 5]