torch.nn.utils.prune.is_pruned

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

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

Параметры

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

Возвращает

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

Примеры

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