torch.nn.utils.prune.remove

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

從模組中移除剪枝重新參數化 (pruning reparameterization)，並從前向鉤子 (forward hook) 中移除剪枝方法 (pruning method)。

名為 name 的已剪枝參數會永久保持剪枝狀態，且名為 name+'_orig' 的參數會從參數列表中移除。 類似地，名為 name+'_mask' 的緩衝區也會從緩衝區列表中移除。

注意

剪枝本身**不會**被取消或反轉！

參數

• module (nn.Module) – 包含要剪枝的張量的模組

• name (str) – module 中將執行剪枝的參數名稱。

範例

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