yolov5_s_qat_dis.yml 552 B

1234567891011121314151617181920212223242526272829
  1. Global:
  2. reader_config: configs/yolov5_reader.yml
  3. include_nms: True
  4. Evaluation: True
  5. model_dir: ./yolov5_s_300e_coco
  6. model_filename: model.pdmodel
  7. params_filename: model.pdiparams
  8. Distillation:
  9. alpha: 1.0
  10. loss: soft_label
  11. QuantAware:
  12. use_pact: true
  13. activation_quantize_type: 'moving_average_abs_max'
  14. quantize_op_types:
  15. - conv2d
  16. - depthwise_conv2d
  17. TrainConfig:
  18. train_iter: 3000
  19. eval_iter: 1000
  20. learning_rate: 0.00001
  21. optimizer_builder:
  22. optimizer:
  23. type: SGD
  24. weight_decay: 4.0e-05
  25. target_metric: 0.365