-
Hi all ,
Please how can i use channel pruning in my model YOLOX (object detection) ?
thanks in advance
-
Object detection models from torchvision not working. If you try to prune pretrained object detection models from [torchvision](https://pytorch.org/vision/stable/models.html#object-detection) you will…
-
Hello,
I am trying to prune [yolov5](https://github.com/ultralytics/yolov5) model using Torch_Pruning. It fails with error message:
KeyError: Conv2d(32, 64, kernel_size=(3, 3), stride=(2, 2), padd…
-
Hi @madlag @julien-c @co42 @srush @Narsil
I am trying to use `nn_pruning` for Pruning different transformer models.
Code:
```
model_checkpoint = "t5-small"
t5small_model = AutoModelForSeq2…
-
您好,我明明安装了pip install torch-pruning,但是还是再pruning_plan = DG.get_pruning_plan((model.backbone.backbone.dark2)[0].dconv.conv, tp.prune_conv, idxs=pruning_idxs)这个函数运行时候出现
AttributeError: module 'torch_pru…
-
I'm trying to prune [DiT-XL-2-256](https://huggingface.co/facebook/DiT-XL-2-256) with taylor pruning.
While pruning the model, the following error occurs during `pruner.step()`.
```
import torch
…
-
Hello
Please, I have a problem concerning the optimization of YOLOX model by Pytorch Pruning (Global & Local)
I got the same inference time after the optimization, I don't understand why ?
…
-
https://virtual2023.aclweb.org/paper_P5570.html#slides
-
https://www.tensorflow.org/model_optimization/guide/pruning/pruning_with_keras
-
Hi, thanks for the great work. I have a question about the experiments in predefined structured pruning methods. I am not sure I am understanding the paper correctly.
For predefined structured prun…