search

marcoancona / TorchPruner

On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for neural networks in PyTorch.
https://arxiv.org/abs/2006.01795
MIT License
162 stars 20 forks source link

Global pruning #6

Closed Christian-lyc closed 18 hours ago

Christian-lyc commented 1 month ago

Hi, I want to ask does it support global structure pruning. Does it support ultralytics yolov8 model? Thank you

marcoancona commented 1 month ago

Hi @Christian-lyc, unfortunately I no longer have capacity to work on this library at the moment. I would not be able to say whether all types of layer of ultralytics yolov8 are supported, I am sorry.