-
Hi @VainF ! Thank you for your work, but I have some strange problems:
After pruning on my own model, the weight of the model does not match any more.
(RuntimeError: Given groups=1, weight of size […
-
hi, i came up with this error when finetuning the yolov6 model:
... in _trace_computational_graph
module2node[param].pruning_dim = dim
KeyError: Parameter containing:
tensor([0., 1., 2., 3., 4.…
-
Hi,
I'm curious to know whether the quantized version of networks are supported, as today I tried that and faced this issue :
```python
QuantizedResnet18 took 35.105 ms [min/max: 35.1/35.1] ms for…
-
When I use this YOLOv8 pruning routine, I cannot carry out deepcopy, I use OBB model, I suspect it is caused by hook operation, but I do not know how to solve it.
BUG:Only Tensors created explicitly …
-
### Search before asking
- [X] I have searched the Ultralytics YOLO [issues](https://github.com/ultralytics/ultralytics/issues) and [discussions](https://github.com/ultralytics/ultralytics/discussi…
-
Hi @VainF,
After pruning, conv -> linear, shouldn't the shape for the linear be (8,12) instead of (8, 15) ? Here is a minimal working example to reproduce:
```python=
import sys, os
sys.path.a…
-
I am attempting to perform pruning for ppipnet (https://github.com/jhb86253817/PIPNet). I used version 1.1.19 of torch pruning, but I encountered an error. How can I correct it? Here is my code:
```
…
-
Thank you for the excellent research.
I have a question about the cfgs file provided in the code. When I tried to reproduce using the [cbgs_voxel0075_voxelnext.yaml](https://github.com/dvlab-researc…
-
I've tried putting the pruning code into train.py, but its detection is only half as good as the original, is there a better way? thanks
-
The memory of my pytorch model increases after I save it to my directory using torch.save(). Also, the inference of my model does not really speed up. Shouldn't it decrease the memory and increase inf…