picodet_s_qat_dis.yaml 636 B

12345678910111213141516171819202122232425262728293031323334
  1. Global:
  2. reader_config: ./configs/picodet_reader.yml
  3. include_nms: True
  4. Evaluation: True
  5. model_dir: ./picodet_s_416_coco_npu/
  6. model_filename: model.pdmodel
  7. params_filename: model.pdiparams
  8. Distillation:
  9. alpha: 1.0
  10. loss: l2
  11. QuantAware:
  12. use_pact: true
  13. activation_quantize_type: 'moving_average_abs_max'
  14. weight_bits: 8
  15. activation_bits: 8
  16. quantize_op_types:
  17. - conv2d
  18. - depthwise_conv2d
  19. TrainConfig:
  20. train_iter: 8000
  21. eval_iter: 1000
  22. learning_rate:
  23. type: CosineAnnealingDecay
  24. learning_rate: 0.00001
  25. T_max: 8000
  26. optimizer_builder:
  27. optimizer:
  28. type: SGD
  29. weight_decay: 4.0e-05