25 lines
782 B
YAML
25 lines
782 B
YAML
_BASE_: [
|
|
'../../yolov3/yolov3_r34_270e_coco.yml',
|
|
]
|
|
|
|
pretrain_weights: https://paddledet.bj.bcebos.com/models/yolov3_r34_270e_coco.pdparams
|
|
|
|
slim: DistillPrune
|
|
|
|
distill_loss: DistillYOLOv3Loss
|
|
|
|
DistillYOLOv3Loss:
|
|
weight: 1000
|
|
|
|
pruner: Pruner
|
|
|
|
Pruner:
|
|
criterion: l1_norm
|
|
pruned_params: ['conv2d_27.w_0', 'conv2d_28.w_0', 'conv2d_29.w_0',
|
|
'conv2d_30.w_0', 'conv2d_31.w_0', 'conv2d_32.w_0',
|
|
'conv2d_34.w_0', 'conv2d_35.w_0', 'conv2d_36.w_0',
|
|
'conv2d_37.w_0', 'conv2d_38.w_0', 'conv2d_39.w_0',
|
|
'conv2d_41.w_0', 'conv2d_42.w_0', 'conv2d_43.w_0',
|
|
'conv2d_44.w_0', 'conv2d_45.w_0', 'conv2d_46.w_0']
|
|
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]
|