-
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…
-
### Checklist
- I have searched related issues but cannot get the expected help.
- I have read related documents and don't know what to do.
### Describe the question you meet
\[here\]
###…
-
**Describe the issue**:
**Cannot speed up depthwise separable convolutions**
[2022-05-16 11:22:40] INFO (nni.compression.pytorch.speedup.compressor/MainThread) replace module (name: _backbone._block…
-
您好,请问在运行基于Torch-Pruning实现MagnitudePruner的实验中,出现如上报错该怎么解决?谢谢。似乎是torch_pruning代码里面的错误,实在不知道如何解决
---> [19] pruner = tp.pruner.MetaPruner(
[20] model,
[2…
-
**Describe the issue**:
When I run `SlimPruner` on ResNet in dependency_aware mode, I am getting this error:
```
File "/home/ubuntu/....", line 67, in init_pruner
pruner.compress()
File …
-
We need some kind of AI to play with and evaluate our agent after training. There are many algorithms we can implement and in this project, we can try the **MiniMax** algorithm with and without _alpha…
-
I try to prune yolox from [https://github.com/Megvii-BaseDetection/YOLOX](url)
```
pruner = tp.pruner.MagnitudePruner(
model,
example_inputs,
importance=imp,
iterative_st…
-
我的剪枝器设置:
example_inputs = torch.randn(1, 3, 640, 640).to(device)
ignored_layers = []
unwrapped_parameters = []
importance = tp.importance.GroupNormImportance(p=2)
pruner = tp.…
-
I'm trying to run this script
https://github.com/VainF/Torch-Pruning/tree/master/examples/yolov8
on segmentation model (yolov8n-seg). I added this lines at the start of prune()
```python
prun…
-
Is it possible to do neural network topology optimization with Pybrain?
For example network pruning such as optimal brain damage or finding the optimal topology with genetic algorithms?
If this is no…