12345678910111213141516171819 |
- _BASE_: [
- '../../retinanet/retinanet_r101_fpn_2x_coco.yml',
- ]
- pretrain_weights: https://paddledet.bj.bcebos.com/models/retinanet_r101_fpn_2x_coco.pdparams
- slim: Distill
- slim_method: FGD
- distill_loss: FGDFeatureLoss
- distill_loss_name: ['neck_f_4', 'neck_f_3', 'neck_f_2', 'neck_f_1', 'neck_f_0']
- FGDFeatureLoss:
- student_channels: 256
- teacher_channels: 256
- temp: 0.5
- alpha_fgd: 0.001
- beta_fgd: 0.0005
- gamma_fgd: 0.0005
- lambda_fgd: 0.000005
|