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')
本页目录