Question: Why is Eigendamage and FPGM not docummented nor exposed in importance.py

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

Question: Why is Eigendamage and FPGM not docummented nor exposed in importance.py #361

Open Hrayo712 opened 3 months ago

Hrayo712 commented 3 months ago

Hi @VainF ,

first of all, thanks for the great work!

Looking at importance.py I noticed that there are 2 importances which are not documented in the docs, nor exposed, that is, FPGMImportance and OBDCImportance.

I was wondering, is there a reason why ? maybe they are under development or not well tested ?

Thanks in advance for the clarification!

VainF commented 3 months ago

Hi @Hrayo712, these two criteria were added recently by @HollyLee2000. We need more evaluation results for them.

An example can be found here: