yolov3_mobilenet_v1_coco_distill_prune.yml 782 B

123456789101112131415161718192021222324
  1. _BASE_: [
  2. '../../yolov3/yolov3_r34_270e_coco.yml',
  3. ]
  4. pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_r34_270e_coco.pdparams
  5. slim: DistillPrune
  6. distill_loss: DistillYOLOv3Loss
  7. DistillYOLOv3Loss:
  8. weight: 1000
  9. pruner: Pruner
  10. Pruner:
  11. criterion: l1_norm
  12. pruned_params: ['conv2d_27.w_0', 'conv2d_28.w_0', 'conv2d_29.w_0',
  13. 'conv2d_30.w_0', 'conv2d_31.w_0', 'conv2d_32.w_0',
  14. 'conv2d_34.w_0', 'conv2d_35.w_0', 'conv2d_36.w_0',
  15. 'conv2d_37.w_0', 'conv2d_38.w_0', 'conv2d_39.w_0',
  16. 'conv2d_41.w_0', 'conv2d_42.w_0', 'conv2d_43.w_0',
  17. 'conv2d_44.w_0', 'conv2d_45.w_0', 'conv2d_46.w_0']
  18. pruned_ratios: [0.5,0.5,0.5,0.5,0.5,0.5,0.7,0.7,0.7,0.7,0.7,0.7,0.8,0.8,0.8,0.8,0.8,0.8]