yolov7_l_qat_dis.yaml 579 字节
Newer Older
1 2

Global:
W
Wenyu 已提交
3
  reader_config: configs/yolov5_reader.yml
4
  include_nms: True
5 6 7 8 9 10 11 12 13
  Evaluation: True
  model_dir: ./yolov7_l_300e_coco
  model_filename: model.pdmodel
  params_filename: model.pdiparams

Distillation:
  alpha: 1.0
  loss: soft_label

14
QuantAware:
15 16 17 18 19 20
  activation_quantize_type: 'moving_average_abs_max'
  quantize_op_types:
  - conv2d
  - depthwise_conv2d

TrainConfig:
W
Wenyu 已提交
21
  train_iter: 8000
22
  eval_iter: 1000
W
Wenyu 已提交
23 24
  learning_rate: 
    type: CosineAnnealingDecay 
25
    learning_rate: 0.00003
W
Wenyu 已提交
26
    T_max: 8000
27
  optimizer_builder:
W
Wenyu 已提交
28
    optimizer:
29
      type: SGD
W
Wenyu 已提交
30
    weight_decay: 0.00004