retinanet_resnet101_coco_distill.yml 382 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
_BASE_: [
  '../../retinanet/retinanet_r101_fpn_2x_coco.yml',
]

pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet101_pretrained.pdparams

slim: Distill
slim_method: FGD
distill_loss: FGDFeatureLoss

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