mask_rcnn_r50_fpn_1x_qat.yml 579 B

12345678910111213141516171819202122
  1. pretrain_weights: https://paddledet.bj.bcebos.com/models/mask_rcnn_r50_fpn_1x_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: 5
  10. LearningRate:
  11. base_lr: 0.001
  12. schedulers:
  13. - !PiecewiseDecay
  14. gamma: 0.1
  15. milestones: [3, 4]
  16. - !LinearWarmup
  17. start_factor: 0.001
  18. steps: 100