-
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 …
-
Is there any way to use tfmot for Keras Yolov3-v4 model pruning, or any other object detection model.
-
Dear @AlexeyAB ,
I design a network and train it on my custom dataset by using darknet framework.
My network weight is about 280MB. I want to use Model compression technology
(Model Pruning, K…
-
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…
-
hi, when I run prune.py, then it just print the model summary, threshold and got stuck, nothing happen.
"YOLOv8-samt_multi3 summary: 321 layers, 9792692 parameters, 0 gradients
tensor(1.0654)"
…
-
Thanks for your work, but when I run this code, there is a error: 'Namespace' object has no attribute 'use_kwta', in utils/continual_model.py", line 244, in structured_pruning.
-
After using the pruning API on the model, when I try to perform the speed up, it throws this error :
**Error: TypeError: forward() missing 1 required positional argument: input**
Traceback:
`m…
-
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…
-
当我用--learner=chn-pruned-rmt \
--cpr_prune_ratio=0.40
压缩自定义模型的时候,发现无论我设置的cpr_prune_ratio为多少,最后压缩完成的模型的参数量都是一样的。
-
**MacOS**
- TensorFlow version (you are using): 2.3
- Are you willing to contribute it (Yes/No): No
**Motivation**
The document just shows how to prune the model during training. Each time, …