I have problem in doing channels wise pruning

VainF / Torch-Pruning

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

https://arxiv.org/abs/2301.12900

MIT License

2.59k stars 318 forks source link

I have problem in doing channels wise pruning #246

Open JongWoongseo opened 1 year ago

JongWoongseo commented 1 year ago

As layer have channels 32 size, if useless channels index are 32, this layer can't pruned. But i want depth wise pruning too. how to solve this problem?

VainF commented 1 year ago

Hi @JongWoongseo, I'm not very familiar with the depth-wise pruning. Could you provide more details about it?