search

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

请问用AB版yolov3训练出来的.weight模型文件，如何在本工程剪枝 #147

Open Edison0902 opened 3 years ago

Edison0902 commented 3 years ago

我使用的是自定义数据集（4类），然后发现重新按照您的工程正常训练200epoch后loss大于20、map不足0.1. 在AB版yolov3训练了50200iters后loss小于1.5、map0.28. 想请教一下： （1）是否是我对本工程使用不当？是否训练的epoch数太少？ （2）请问能否直接加载AB版yolov3训练出来的.weight模型文件进行剪枝，如何操作？

`U_LG{1D4NZKY05Q28M5UWP Z$V)F~GV$GIVGIU0JBP`$WM

zbyuan commented 3 years ago

1、pytorch版本的YOLOv3 跟darknet下的v3在实现的方式都略有不同，在pytorch下加载预训练权重。 2、损函略有不同，建议依照readme进行

ForrestHeiYing commented 3 years ago

@zbyuan AB版模型可以用本工程直接稀疏训练和剪枝吗？搞定没有呢