问题描述
我正在通过以下方式加载 torchscript 模型:
model = torch.jit.load("model.pt").to(device)
此模型的子模块标识为 RecursiveScriptModule
。我想微调上传的权重,以使其更简单并将它们投射到 torch.float32
最好将所有这些东西转换为普通的 PyTorch nn.Module
。>
在官方文档 https://pytorch.org/docs/stable/jit.html 中,它被告知如何将 nn.Module
转换为 torchscript
,但我没有找到任何相反方向的示例。有没有办法做到这一点?
P.S 这里给出了加载模型预训练模型的例子:https://github.com/openai/CLIP/blob/main/notebooks/Interacting_with_CLIP.ipynb
解决方法
您可以尝试按原样加载它,例如state_dict = torch.load(src).state_dict()
。
然后手动转换每个键和值 new_v = state_dict[k].cpu().float()
。