-
Hi all and thank you for your amazing work!
I managed easily to use your yolov8 example in order to train and prune it and it works like a charm, managed to load it for inference after and I gain 2…
-
When I first demonstrated using Kan for a demo, the actual execution of the code was not the same as in the sample.
I found that when constructing nodes, some nodes can also be trimmed.
I don't know…
plumv updated
1 month ago
-
Before Pruning: MACs=1.009904 G, #Params=0.007226 G
After Pruning: MACs=0.379899 G, #Params=0.002635 G
image 1/128 D:\Torch-Pruning-master\data\coco128\images\train2017\000000000009.jpg: 480x640 (no…
-
I am trying to prune the [Restormer](https://github.com/swz30/Restormer) model using Torch-pruning library but I stumbled upon the following error :
```
File "/home/sarthak.vora/Documents/pt-text…
-
Hello, recently I have been applying your code to yolov8. Based on previous problems encountered, and the help of the files in the examples/yolov8 folder. I managed to reduce both the number of parame…
-
Hello,
I'm attempting to train a model for a micro-controller that only supports 8-bit precision or lower.
This works perfectly when training using your `QuantizationAwareTrainingConfig`.
In add…
-
May I ask if this tool is currently unable to perform pruning on GQA models? Llama2-70B or Llama3
-
Hi. I am trying to prune MViTv2 with random pruning with the following code
```python
import torch
import timm
from timm.models.mvitv2 import MultiScaleAttention, MultiScaleVit as MViT
import tor…
-
Is there any way to use tfmot for Keras Yolov3-v4 model pruning, or any other object detection model.
-
Hello,
I followed the example and come up with this:
```
model = resnet18(pretrained=True).eval()
example_inputs = torch.randn(1, 3, 224, 224)
DG = tp.DependencyGraph().build_dependency(mod…