yezechen

> 大佬您好，谢谢你的开源，请问我用yolov5m完成稀疏训练过后，在用prune.py剪枝的过程中出现了以下报错，希望大佬解答一下在m，l模型剪枝的时候需要对prune.py做哪些修改，还是我这里是其他错误的原因，十分感谢感谢感谢 from n params module arguments 0 -1 1 5280 models.common.Focus [3, 48, 3] 1 -1 1 41664 models.common.Conv [48, 96, 3, 2] 2 -1 1 42048 models.pruned_common.C3Pruned [96,...

请问楼主解决了吗，我也遇到了这个问题

> > 请问楼主解决了吗，我也遇到了这个问题 > > 你尝试一下把sr再缩小10倍再训练 我试一下，谢谢

> 换成Adam优化器 训练时候参数已经设置了adam

> 好像也不是用的6.0，可能是因为我改动过SPP？ > […](#) > ------------------ 原始邮件 ------------------ 发件人: msddmzy ***@***.***> 发送时间: 2022年3月5日 21:56 收件人: midasklr/yolov5prune ***@***.***> 抄送: weinanyu ***@***.***>, Author ***@***.***> 主题: Re: [midasklr/yolov5prune] 剪枝时报错：KeyError: 'model.0.conv.bn' (Issue #74) 试了很多的稀疏度和剪枝率都报这个同样的错，不知如何解决...

> you should modify include/ctdetConfig.h to fix it how to fix it？