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

跳闸 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