Torch-Pruning
Torch-Pruning copied to clipboard
YOLOv5剪枝后推理无结果
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结构进行剪枝后推理就无结果,如何控制剪枝的过程?