请问代码支持yolo5m6、l6模型的剪枝吗

midasklr / yolov5prune

548 stars 113 forks source link

Open yazheng0307 opened 2 years ago

midasklr commented 2 years ago

理论上是支持的，之前没有实验过，yolov5l6模型并没有特殊的操作

WZJAI2018 commented 2 years ago

请问你解决了嘛，修改成功能剪枝yolo5m6模型了嘛

yazheng0307 commented 2 years ago

请问你解决了嘛，修改成功能剪枝yolo5m6模型了嘛

直接对m6剪枝会有key找不到报错，我用的方法是直接跳过没有找到的层

zxsitu commented 2 years ago

@yazheng0307 您好，请问可以分享修改后的代码吗？