torch.nn.utils.prune.remove¶ torch.nn.utils.prune.remove(module, name)[源代码]¶ 从模块中移除剪枝重参数化，并从前向钩子中移除剪枝方法。 名为 name 的修剪参数将永久修剪，并且名为 name+'_orig' 的参数将从参数列表中移除。同样，名为 name+'_mask' 的缓冲区也将从缓冲区中移除。 注意 剪枝本身不会被撤销或逆转！ Parameters 模块 (nn.Module) – 包含要剪枝的张量的模块 名称 (字符串) – 在 模块 中进行剪枝操作的参数名称。 示例 >>> m = random_unstructured(nn.Linear(5, 7), name='weight', amount=0.2) >>> m = remove(m, name='weight')
