torch.nn.utils.prune.remove¶
- torch.nn.utils.prune.remove(module, name)[source][source]¶
从模块中移除修剪重新参数化以及从前向钩子中移除修剪方法。
被修剪的参数
name将永久修剪，而参数name+'_orig'将从参数列表中移除。同样，缓冲区name+'_mask'也将从缓冲区中移除。注意
自剪枝操作不可撤销或逆转！
- 参数:
module (nn.Module) – 包含要剪枝的张量的模块
名称（str）- 在
module中参数名称，剪枝将在此参数上执行。
示例
>>> m = random_unstructured(nn.Linear(5, 7), name='weight', amount=0.2) >>> m = remove(m, name='weight')
