文件名称：Pruning-PFF:NeurIPS 2020 论文“Pruning Filter in Filter”的 PyTorch 实现
文件大小：151KB
文件格式：ZIP
更新时间：2024-06-19 04:58:33
pruning neurips-2020 Python
在过滤器中修剪过滤器 介绍 这是我们的 NeurIPS 2020 论文“Pruning Filter in Filter”的 PyTorch 实现。 在本文中： 我们提出了一种新的剪枝范式，称为 Stripe-wise-Pruning(SP)，它可以看作是 Filter-Pruning (FP) 的一般情况。 SP 将过滤器 $F \in \mathbb{R}^{C\times K\times K}$ 视为 $K\times K$ 条纹（即 $1\times 1$ 过滤器 $\in \mathbb{R}^ c $ )，并以条带为单位而不是整个过滤器进行修剪。 与现有方法相比，SP 实现了比传统 FP 更精细的粒度，同时比 Weight-Pruning 对硬件友好，并且比 Group-wise-Pruning 保持过滤器之间的独立性，从而在 CIFAR-10 和 ImageNet 上实现最
【文件预览】：
Pruning-PFF-main
----README.md(6KB)
----LICENSE(1KB)
----.gitattributes(66B)
----BrokenNet_filter.png(154KB)
----models()
--------stripe.py(5KB)
--------resnet56.py(3KB)
--------__init__.py(65B)
--------vgg.py(2KB)
----main.py(6KB)
----flops.py(3KB)