torch.nn.utils.prune.is_pruned — PyTorch 2.7 documentation (original) (raw)

torch.nn.utils.prune.is_pruned(module)[source][source]¶

Check if a module is pruned by looking for pruning pre-hooks.

Check whether module is pruned by looking forforward_pre_hooks in its modules that inherit from theBasePruningMethod.

Parameters

module (nn.Module) – object that is either pruned or unpruned

Returns

binary answer to whether module is pruned.

Examples

from torch.nn.utils import prune m = nn.Linear(5, 7) print(prune.is_pruned(m)) False prune.random_unstructured(m, name='weight', amount=0.2) print(prune.is_pruned(m)) True