torch.nn.utils.prune.is_pruned

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

透過查詢剪枝的 pre-hooks 來檢查模組是否已被剪枝。

透過查詢其繼承自 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