-
Hi. Thank you for your contribution.
Can OTO be used for pruning Transformers? I want to apply it to a DETR-like object detection model(DINO:https://github.com/IDEA-Research/DINO). Is this feasible?…
-
Hello, thanks for the nice work.
One question, I believe the code and process of model pruning is not included in the current training and eval script, is this correct?
Thanks!
-
Thank you for your project. I noticed the presence of the mlp_mixer models in the project. I am curious if sparsification has been implemented on this model or on other image classification models.
-
Hi, thank you for sharing. I found that after pruning, the model size of slimsam-77 is only 38M, which is the same as the model size of edgesam, but the GPU usage of slimsam is still very high, 3071Mi…
-
Dear Authors,
Thank you for sharing the amazing and inspiring pruning work!
As mentioned in [celebahq-ldm-vq-4.yaml](https://github.com/VainF/Diff-Pruning/blob/main/ldm_exp/configs/latent-diffusi…
-
It is the test code:
```
import torch
import torch_pruning as tp
import torch.nn as nn
import clip.clip as clip
clip_model = clip.load('ViT-B/32', device='cpu', jit=False, )[0]
model = clip_m…
-
when I pruner https://github.com/chenjun2hao/DDRNet.pytorch
I meet the error
`
def prune_model(model):
model.cpu()
DG = tp.DependencyGraph().build_dependency(model, torch.randn((1…
-
### Search before asking
- [X] I have searched the YOLOv8 [issues](https://github.com/ultralytics/ultralytics/issues) and [discussions](https://github.com/ultralytics/ultralytics/discussions) and f…
-
Hi thanks for the great code.
Please could you provide clear instructions for pruning a pre-trained INRIA "checkpoint" model .ply
First off, Checkpoints and Trained models seem to be referenced …
-
Hi,
I am trying to use [ModelPruning callback](https://pytorch-lightning.readthedocs.io/en/latest/extensions/generated/pytorch_lightning.callbacks.ModelPruning.html#pytorch_lightning.callbacks.Mode…