picodet_s_quant.yml 653 B

1234567891011121314151617181920212223242526
  1. pretrain_weights: https://paddledet.bj.bcebos.com/models/picodet_s_320_coco.pdparams
  2. slim: QAT
  3. QAT:
  4. quant_config: {
  5. 'activation_preprocess_type': 'PACT',
  6. 'weight_quantize_type': 'channel_wise_abs_max', 'activation_quantize_type': 'moving_average_abs_max',
  7. 'weight_bits': 8, 'activation_bits': 8, 'dtype': 'int8', 'window_size': 10000, 'moving_rate': 0.9,
  8. 'quantizable_layer_type': ['Conv2D', 'Linear']}
  9. print_model: False
  10. epoch: 50
  11. LearningRate:
  12. base_lr: 0.001
  13. schedulers:
  14. - !PiecewiseDecay
  15. gamma: 0.1
  16. milestones:
  17. - 30
  18. - 40
  19. - !LinearWarmup
  20. start_factor: 0.
  21. steps: 100
  22. TrainReader:
  23. batch_size: 96