retinanet_resnet101_coco_distill.yml 457 Bytes
Newer Older
wangkx1's avatar
wangkx1 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
_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