torch.nn.utils.prune.is_pruned

torch.nn.utils.prune.is_pruned(module)

檢查模組是否透過尋找剪枝 pre-hook 進行剪枝。

透過在其繼承自 BasePruningMethod 的模組中尋找 forward_pre_hooks 來檢查 module 是否已剪枝。

參數

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