-
Hello,
I am trying to prune yolov5 model and my env is as follows:
torch==1.8.1
torchvision==0.9.1
nncf==1.7.1
Python 3.6.8
No CUDA
While the sparsity compression works fine, filter pruning doe…
-
Hello again :)
Having tried different approach to make small changes to DivHGNN, I wanted to have an idea of exactly how each part of your design contributes to the very good result highlithed in t…
emduc updated
1 month ago
-
@VainF Thanks for amazing repo.
I tried to run inference for yolov7 pruned model on one image, **but pruned model did not detect anything** (output image without bboxes) https://github.com/VainF/Torc…
-
Hi, thank you for sharing the amazing and inspiring pruning work! We have a question related to the pruning code for LDM.We are confused why in the [prune_ldm.py line 120](https://github.com/VainF/Dif…
-
Hello, I met some confusions after reading your paper, hope to get the answers, thank you very much!
1. The "pruning" operation in the code seems to be just setting the weight paramete…
-
I try to follow the example to prune the transformer model, but the layer-norm always gets a size mismatch error.
-
## 🐛 Bug
Hi, I was working on iteratively pruning with global_pruning method (training --> prune --> training --> prune). However, I found it costs GPU memory after pruning.
With a very big mode…
-
My scripts:
```
import torch
from mmpretrain import get_model
import torch_pruning as tp
import torch.nn as nn
from typing import Sequence
from mmpretrain.models.utils import PatchMerging, Wind…
-
**Describe the bug**
Trying to save a model that wraps a `MultiHeadAttention` layer in a `PruneLowMagnitude`, fails with duplicate dataset name.
**System information**
TensorFlow version (inst…
-
If we can distill or prune the NLLB-200 shortly after starting fine tuning, we may be able to dramatically reduce (50% or more) the training and inferencing time needed. It could even do something lik…