【文件属性】:
文件名称:flop:Pytorch库用于基于L0的因式修剪
文件大小:102KB
文件格式:ZIP
更新时间:2021-05-17 17:28:31
Python
跳闸
Pytorch库,用于基于L0的修剪,如本文所述:
安装
pip install -U flop
用法
创建尺寸为N的硬混凝土面罩:
from flop import HardConrete
N = 100
hardconcrete = HardConcrete ( n_in = N )
然后,您可以使用以下方法即时采样口罩:
mask = hardconcrete ()
请注意,在评估期间,会编译并修复一个遮罩。
您可能还会发现这些其他对象很有用:
ProjectedLinear :替换线性层以包括中间投影。
HardConreteProjectedLinear : ProjectedLinear模块的硬版本。
您可以直接实例化HardConcrete对象,也可以选择首先使用ProjectedLinear模块进行训练,然后通过以下方式引入hardconcret
【文件预览】:
flop-master
----.github()
--------workflows()
----LICENSE(1KB)
----requirements.txt(68B)
----examples()
--------enwik8()
--------__init__.py(0B)
--------enwik8_tf()
--------wt103()
----setup.py(993B)
----README.md(2KB)
----flop()
--------agp.py(3KB)
--------linear.py(25KB)
--------utils.py(7KB)
--------hardconcrete.py(4KB)
--------__init__.py(1KB)
--------embedding.py(26KB)
--------version.py(74B)
----.flake8(81B)