Torch-Pruning icon indicating copy to clipboard operation
Torch-Pruning copied to clipboard

YOLOv5剪枝后推理无结果

Open hhitdata opened this issue 1 year ago • 0 comments

from models.yolo import Detect, SPPF, Concat, C3, Conv for m in model.model.modules(): if isinstance(m, Detect) or isinstance(m, Conv): ignored_layers.append(m)

经过尝试后发现如果将Conv结构进行剪枝后推理就无结果,如何控制剪枝的过程?

hhitdata avatar Aug 29 '24 09:08 hhitdata