search

VainF / Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs
https://arxiv.org/abs/2301.12900
MIT License
2.61k stars 323 forks source link

module2node bug #309

Open QiqLiang opened 9 months ago

QiqLiang commented 9 months ago

hi, i came up with this error when finetuning the yolov6 model: ... in _trace_computational_graph module2node[param].pruning_dim = dim KeyError: Parameter containing: tensor([0., 1., 2., 3., 4., 5., 6., 7., 8., 9., 10., 11., 12., 13., 14., 15., 16.])

why does it happen and how can i solve it? thanks a lot for answering and it would be better if you can write a yolov6_pruning.py！

ganliqiang commented 8 months ago

i meet the same error have you solved it already?

xiaowenhe commented 6 months ago

@QiqLiang @ganliqiang , Does solved it? If solved, How to solve?