-
when I use 'Torch-Pruning/examples/coco_yolo/train_with_pruning.py' to pure yolov5s , will have error'RuntimeError: Given groups=1, weight of size [255, 128, 1, 1], expected input[8, 77, 80, 80] to…
-
Got
```
RuntimeError: Given groups=1, weight of size [255, 128, 1, 1], expected input[22, 77, 80, 80] to have 128 channels, but got 77 channels instead
```
when running
```
python train_with_pru…
-
- ShuffleNet
- [Channel Pruning](https://github.com/yihui-he/channel-pruning)
-
**System information**
- TensorFlow version (you are using): 2.5.0
- Are you willing to contribute it (Yes/No): Yes
**Motivation**
Deciding on where to have high filter/channel counts in con…
-
Have you completed this project? I haven't seen the pruning channels codes....
Do you plan to actually prune the channels from the network?
-
python layer_channel_prune.py --cfg cfg/my_cfg.cfg --data data/my_data.data --weights weights/last.pt --shortcuts 12 --global_percent 0.8 --layer_keep 0.1
Traceback (most recent call last):
File …
-
Thanks for your great work! I noticed that different criteria of channel pruning and token pruning on DeiT-Small are compared in Table 3.
![image](https://user-images.githubusercontent.com/43805318/2…
-
Dear Author(s),
I am trying to applying this channel pruning method on a Unet(which is RCNN-like network with residual branch. I have tried weeks to read your python code (Also I am quite familar…
-
You have two dead links in your Wiki. In Quick Start -> 3. High-level Prunes -> Pruning with High-level Pruners -> "this tutorial" and "benchmarks/main.py"
#### Pruning with High-level Pruners
L…
-
Hi, thanks for contributing a great library!
I've been doing a close-up study of the `MultiheadAttentionPruner` implementation, and I have some concerns.
The pruning of output channel in out_pro…