prune_by_desc

功能说明

根据已有的剪枝信息，在推理时对模型进行剪枝。

函数原型

prune_by_desc(desc)

参数说明

调用示例

from msmodelslim.pytorch.prune.prune_torch import PruneTorch
model = torchvision.models.vgg16(pretrained=False)
model.eval()
prune_torch = PruneTorch(model, torch.ones([1, 3, 224, 224]).type(torch.float32))
desc = prune_torch.prune()
prune_torch.prune_by_desc(desc)