flop:Pytorch库用于基于L0的因式修剪

时间:2024-06-06 11:15:11
【文件属性】:

文件名称:flop:Pytorch库用于基于L0的因式修剪

文件大小:102KB

文件格式:ZIP

更新时间:2024-06-06 11:15:11

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)

网友评论