torch.nn.utils.prune.remove
- torch.nn.utils.prune.remove(module, name)[源代码]
-
从模块中删除剪枝重参数化,并从前向钩子中删除剪枝方法。
名为
name
的剪枝参数将永久保持被剪枝状态,而名为name+'_orig'
的参数将从参数列表中删除。同样地,名为name+'_mask'
的缓冲区也将从缓冲区列表中移除。注意
剪枝操作本身无法撤销或逆转!
示例
>>> m = random_unstructured(nn.Linear(5, 7), name='weight', amount=0.2) >>> m = remove(m, name='weight')