时间戳

首页 > 解决方案 > 修剪模型后删除 Pytorch 中的 weight_orig

deep-learning - 修剪模型后删除 Pytorch 中的 weight_orig

问题描述

在 Pytorch 中修剪模型后，保存的模型包含修剪后的权重和 weight_orig。这会导致修剪后的模型大小大于未修剪的模型。有没有办法删除 weight_orig 并减小修剪后的模型大小？

标签： deep-learningpytorchpruning

解决方案

如官方文档中所述，您可以将torch.nn.utils.prune.remove()其用于此目的。删除根据和
remove()的重新参数化，并删除. 你会这样使用它：weight_origweight_maskforward_pre_hook

for module in model.modules():
    if isinstance(module, torch.nn.Conv2d):
        prune.remove(module,'weight')
    # etc...

推荐阅读