yolov6mt_s_qat_dis.yaml 582 B

123456789101112131415161718192021222324252627282930
  1. Global:
  2. reader_config: configs/yolov5_reader.yml
  3. include_nms: True
  4. Evaluation: True
  5. model_dir: ./yolov6mt_s_400e_coco
  6. model_filename: model.pdmodel
  7. params_filename: model.pdiparams
  8. Distillation:
  9. alpha: 1.0
  10. loss: soft_label
  11. QuantAware:
  12. activation_quantize_type: 'moving_average_abs_max'
  13. quantize_op_types:
  14. - conv2d
  15. - depthwise_conv2d
  16. TrainConfig:
  17. train_iter: 8000
  18. eval_iter: 1000
  19. learning_rate:
  20. type: CosineAnnealingDecay
  21. learning_rate: 0.00003
  22. T_max: 8000
  23. optimizer_builder:
  24. optimizer:
  25. type: SGD
  26. weight_decay: 0.00004