torch.nn.utils.prune.is_pruned

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

通过查找修剪预钩子来检查模块是否已修剪。

通过查看其模块中的 forward_pre_hooks（这些模块继承自 BasePruningMethod）来检查 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