retinanet_resnet101_coco_distill.yml 457 B

12345678910111213141516171819
  1. _BASE_: [
  2. '../../retinanet/retinanet_r101_fpn_2x_coco.yml',
  3. ]
  4. pretrain_weights: https://paddledet.bj.bcebos.com/models/retinanet_r101_fpn_2x_coco.pdparams
  5. slim: Distill
  6. slim_method: FGD
  7. distill_loss: FGDFeatureLoss
  8. distill_loss_name: ['neck_f_4', 'neck_f_3', 'neck_f_2', 'neck_f_1', 'neck_f_0']
  9. FGDFeatureLoss:
  10. student_channels: 256
  11. teacher_channels: 256
  12. temp: 0.5
  13. alpha_fgd: 0.001
  14. beta_fgd: 0.0005
  15. gamma_fgd: 0.0005
  16. lambda_fgd: 0.000005