剪枝yolov8模型报错统计参数奔溃 concat算子报错

VainF / Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs

https://arxiv.org/abs/2301.12900

MIT License

2.44k stars 308 forks source link

Open MapleAura opened 3 months ago

MapleAura commented 3 months ago

测试环境： pytorch == 1.13 or pytorch==1.8 ultralytics==6a2fddf 运行脚本：示例中的yolov8_example 直接异常原因： x类型应当为list而不是一个tensor

应当如何修改