torch.nn.utils.prune.is_pruned

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

Проверяет, подвергнут ли модуль прореживанию, проверяя наличие предварительных хуков прореживания.

Проверяет, подвергнут ли module прореживанию, проверяя наличие forward_pre_hooks в его модулях, наследующих от BasePruningMethod.

Параметры:

module (nn.Module) – объект, подвергнутый или не подвергнутый прореживанию

Возвращает:

логическое значение, указывающее, подвергнут ли module прореживанию.

Тип возвращаемого значения:

bool

Примеры

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