yolov3_darknet_qat.yml 706 B

12345678910111213141516171819202122232425262728293031
  1. pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_darknet53_270e_coco.pdparams
  2. slim: QAT
  3. QAT:
  4. quant_config: {
  5. 'weight_quantize_type': 'channel_wise_abs_max', 'activation_quantize_type': 'moving_average_abs_max',
  6. 'weight_bits': 8, 'activation_bits': 8, 'dtype': 'int8', 'window_size': 10000, 'moving_rate': 0.9,
  7. 'quantizable_layer_type': ['Conv2D', 'Linear']}
  8. print_model: True
  9. epoch: 50
  10. LearningRate:
  11. base_lr: 0.0001
  12. schedulers:
  13. - !PiecewiseDecay
  14. gamma: 0.1
  15. milestones:
  16. - 30
  17. - 45
  18. - !LinearWarmup
  19. start_factor: 0.
  20. steps: 1000
  21. OptimizerBuilder:
  22. optimizer:
  23. momentum: 0.9
  24. type: Momentum
  25. regularizer:
  26. factor: 0.0005
  27. type: L2