Torch-Pruning icon indicating copy to clipboard operation
Torch-Pruning copied to clipboard
verified publisher icon VainF

这个功能是针对模型的某一模块进行剪枝的吗？

Open GeneralJing opened this issue 4 years ago • 2 comments

我跑了下示例代码，发现模型大小没有变化，看了下代码，发现那个比例针对的是某一个模块，那如果要对整个网络进行剪枝是需要，每一个模块单独进行剪枝吗？不能整体进行剪枝吗？

GeneralJing avatar Oct 09 '21 08:10 GeneralJing

我也遇到了这个问题Torch-Pruning/examples/prune_resnet18_cifar10.py 这个是正确的么

Louis24 avatar Mar 07 '22 09:03 Louis24

这个cifar10+resnet的例子 绝对可以正确工作 先跑训练 然后跑剪枝 剪枝这部分代码是先加载原先训练好的模型 剪枝 然后再训练保存的模型比原模型小了一半 而且精度接近 我用mnn测试了下速度 原模型用时10ms 裁剪后用时5ms 这个库非常好用

MatchX avatar Jun 16 '22 09:06 MatchX