yolov5_s_qat_dis.yml 587 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

Global:
  reader_config: configs/yolov5_reader.yml
  input_list: ['image', 'scale_factor']
  arch: YOLO
  Evaluation: True
  model_dir: ./yolov5_s_300e_coco
  model_filename: model.pdmodel
  params_filename: model.pdiparams

Distillation:
  alpha: 1.0
  loss: soft_label

Quantization:
  use_pact: true
  activation_quantize_type: 'moving_average_abs_max'
  quantize_op_types:
  - conv2d
  - depthwise_conv2d

TrainConfig:
W
Wenyu 已提交
23
  train_iter: 3000
24
  eval_iter: 1000
W
Wenyu 已提交
25
  learning_rate: 0.00001
26 27 28 29
  optimizer_builder:
    optimizer:
      type: SGD
    weight_decay: 4.0e-05
W
Wenyu 已提交
30
  target_metric: 0.365