gfl_ld_distill.yml 494 B

12345678910111213141516171819202122232425
  1. _BASE_: [
  2. '../../gfl/gfl_r18vd_1x_coco.yml',
  3. ]
  4. # teacher pretrain model
  5. pretrain_weights: https://paddledet.bj.bcebos.com/models/gfl_r101vd_fpn_mstrain_2x_coco.pdparams
  6. slim: Distill
  7. slim_method: LD
  8. ResNet:
  9. depth: 101
  10. variant: d
  11. norm_type: bn
  12. freeze_at: 0
  13. return_idx: [1,2,3]
  14. num_stages: 4
  15. TrainReader:
  16. batch_transforms:
  17. - PadBatch: {pad_to_stride: 32}
  18. - Gt2GFLTarget:
  19. downsample_ratios: [8, 16, 32, 64, 128]
  20. grid_cell_scale: 8
  21. compute_vlr_region: True