gfl_r101vd_fpn_coco_distill_cwd.yml 398 B

12345678910111213141516
  1. _BASE_: [
  2. '../../gfl/gfl_r101vd_fpn_mstrain_2x_coco.yml',
  3. ]
  4. pretrain_weights: https://paddledet.bj.bcebos.com/models/gfl_r101vd_fpn_mstrain_2x_coco.pdparams
  5. slim: Distill
  6. slim_method: CWD
  7. distill_loss: ChannelWiseDivergence
  8. distill_loss_name: ['cls_f_4', 'cls_f_3', 'cls_f_2', 'cls_f_1', 'cls_f_0']
  9. ChannelWiseDivergence:
  10. student_channels: 80
  11. teacher_channels: 80
  12. tau: 1.0
  13. weight: 5.0