Python PyTorch remove用法及代碼示例

本文簡要介紹python語言中 torch.nn.utils.prune.remove 的用法。

用法:

torch.nn.utils.prune.remove(module, name)

參數：

• module(torch.nn.Module) -包含要修剪的張量的模塊

• name(str) -module 中的參數名稱，將對其進行修剪。

從模塊中刪除修剪重新參數化，從前向鉤子中刪除修剪方法。名為name 的已修剪參數保持永久修剪狀態，名為name+'_orig' 的參數已從參數列表中刪除。同樣，名為name+'_mask' 的緩衝區將從緩衝區中刪除。

注意

修剪本身不會撤消或逆轉！

例子

>>> m = random_unstructured(nn.Linear(5, 7), name='weight', amount=0.2)
>>> m = remove(m, name='weight')