快捷方式

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')

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源