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

How to use torch prunning with tensorflow model? #358

Open trangtv57 opened 3 months ago

trangtv57 commented 3 months ago

tks for very creative idea of torch pruning, But my model is exported from pretrained tensorflow model, it's hard to fine tune or re-trained again. So can you give me idea, how can I use torch-prunning with my model. Because I see, after you prune, You need to finetune again model. I understand that I can convert my tf model to pytorch, than prune model by this lib => I get new model with new architecture, than I can fine tune new model prune-pytorch? right? Can it's work. Sorry if my question is navie. Tks you