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