torch.nn.utils.prune.remove

torch.nn.utils.prune.remove(module, name)

从模块中移除剪枝重参数化以及从前向钩子中移除剪枝方法。

名为 name 的已剪枝参数将永久保持剪枝状态，而名为 name+'_orig' 的参数将从参数列表中移除。类似地，名为 name+'_mask' 的缓冲区将从缓冲区中移除。

注意

剪枝操作本身**不会**被撤销或反转！

参数

• module (nn.Module) – 包含待剪枝张量的模块

• name (str) – 模块中将要进行剪枝操作的参数名称。

示例

>>> m = random_unstructured(nn.Linear(5, 7), name='weight', amount=0.2)
>>> m = remove(m, name='weight')