lishuang

no plan

代码使用torch版本为1.7，具体配置可参考yolov5第四版

检查类别数是否正确。 这个需要调试解决或者找个会调试代码的人。

加速体现在开发版上，显卡上瓶颈在未达到算力顶峰时为数据复制粘贴耗时

我都重新生成cfg文件了，模型结构怎么没变

你说的是yolov5官方提供的非结构化剪枝吧，这里是结构化剪枝。

报错显示的是维度数目不对，项目未使用SyncBatchNorm

[prune_finetune.py](https://github.com/ZJU-lishuang/yolov5-v4/blob/main/prune_finetune.py) Have a good weekend!

稀疏训练效果好坏的指标并不是map，，map是模型性能指标 项目中提供了具体的可复现的实例。可以先跑通示例加深理解

without the parameters，how to define a train action?