tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝,通道剪枝,知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

我使用和您相同的数据集以及参数训练,channel prun的结果相仿,但是layer prun后的map在微调前只有20,可能是什么问题呢? #14

Open px2017 opened 4 years ago

zbyuan commented 4 years ago

@px2017 hi In the course of our experiments, the layer_prune has a great impact on mAP

tanluren commented 4 years ago

你好,刚更新了剪层策略,你可以试试新策略的效果。剪层对于精度的损害的确较大,但是提速明显。精度可以通过微调回升,不同的剪层数也会有不同的效果。另外影响最大的是稀疏训练,稀疏压缩的好,精度也能保持的高。