Python PyTorch remove用法及代码示例

本文简要介绍python语言中 torch.nn.utils.prune.remove 的用法。

用法:

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

参数：

• module(torch.nn.Module) -包含要修剪的张量的模块

• name(str) -module 中的参数名称，将对其进行修剪。

从模块中删除修剪重新参数化，从前向钩子中删除修剪方法。名为name 的已修剪参数保持永久修剪状态，名为name+'_orig' 的参数已从参数列表中删除。同样，名为name+'_mask' 的缓冲区将从缓冲区中删除。

注意

修剪本身不会撤消或逆转！

例子

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