如何将PyTorch中的torchscript模型转换成普通的nn.Module？
How to convert torchscript model in PyTorch to ordinary nn.Module?
我正在通过以下方式加载 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().
我正在通过以下方式加载 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().