torch.nn.utils.prune.is_pruned

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

Проверить, является ли module обрезанным, посмотрев на forward_pre_hooks в его модулях, которые наследуются от BasePruningMethod.

Параметры:

module (nn.Module) – объект, который либо обрезан, либо нет.

Возвращает:

двоичный ответ о том, является ли module обрезанным.

Примеры

>>> 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