torch.nn.utils.prune.remove

torch.nn.utils.prune.remove(module, name) [source]

Remove the pruning reparameterization from a module and the pruning method from the forward hook.

The pruned parameter named name remains permanently pruned, and the parameter named name+'_orig' is removed from the parameter list. Similarly, the buffer named name+'_mask' is removed from the buffers.

Note

Pruning itself is NOT undone or reversed!

Parameters

• module (nn.Module) – module containing the tensor to prune
• name (str) – parameter name within module on which pruning will act.

Examples

>>> m = random_unstructured(nn.Linear(5, 7), name="weight", amount=0.2)
>>> m = remove(m, name="weight")