LiyuyangSWJTU

大佬您好，谢谢你的开源，请问我用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, 48,...

> > 大佬您好，谢谢你的开源，请问我用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...

> 老哥解决了吗，我也是一模一样的问题，Traceback (most recent call last): File "prune.py", line 807, in opt=opt File "prune.py", line 462, in test_prune assert pruned_model_state.keys() == modelstate.keys() AssertionError yezechen大佬的答案把问题解决了