Python PyTorch is_pruned用法及代码示例

本文简要介绍python语言中 torch.nn.utils.prune.is_pruned 的用法。

用法:

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

参数：

module(torch.nn.Module) -被修剪或未修剪的对象

返回：

module 是否被修剪的二进制答案。

通过在继承自 BasePruningMethod 的模块中查找 forward_pre_hooks 来检查 module 是否被修剪。

例子

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