LiyuyangSWJTU

用yolov5m或者yolov5l完成稀疏训练后剪枝报错

8

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