西伯利亚挖土豆 comments

Repositories
Issues
Comments

Results 2 comments of


                                            西伯利亚挖土豆

How to prune for vit model

> Hi @aoji-tjut, please follow the [tests/test_torchvision_models.py](https://github.com/VainF/Torch-Pruning/blob/master/tests/test_torchvision_models.py) for vit pruning. It is slightly different from CNN pruning because the inference of ViT relies on some internal variables like [num_heads](https://github.com/VainF/Torch-Pruning/blob/06c84bacc0feb4f592c98c30f7940a87544cc29b/tests/test_torchvision_models.py#L132) and...

Prunning AttentionPool2D

Hi, I also meet a problem when I try to prune the ModifiedResNet50 in CLIP model, and the issue comes from the attnpool module too. Have you solved your problem?...