如何将PyTorch中的torchscript model转换为普通的nn.Module？

Question

我正在通过以下方式加载torchscript model：

model = torch.jit.load("model.pt").to(device)

此 model 的子模块被标识为RecursiveScriptModule 。 我想微调上传的权重，以使其更简单并将它们转换为torch.float32最好将所有这些东西转换为普通的 PyTorch nn.Module 。

在官方文档https://pytorch.org/docs/stable/jit.html 中，它被告知如何将nn.Module转换为torchscript ，但我没有找到任何相反方向的示例。 有没有办法做到这一点？

PS这里给出了加载model预训练model的示例： https://github.com/openai/CLIP/blob/main.ipynbsInterbook/

Answer 1

您可以尝试加载它，例如state_dict = torch.load(src).state_dict() 。 然后手动转换每个键和值new_v = state_dict[k].cpu().float() 。