Python 如何将PyTorch中的torchscript模型转换为普通的nn.Module?

Python 如何将PyTorch中的torchscript模型转换为普通的nn.Module?,python,pytorch,torchscript,Python,Pytorch,Torchscript,我以以下方式加载torchscript模型: model = torch.jit.load("model.pt").to(device) 此模型的子模块标识为RecursiveScriptModule。我想对上传的权重进行微调,以使其更简单,并将其转换为torch.float32最好将所有这些内容转换为普通PyTorchnn.Module 在官方文件中,有人告诉我如何将nn.Module转换为torchscript,但我没有发现任何相反方向的例子。有办法做到这一点吗 p.S

我以以下方式加载torchscript模型:

model = torch.jit.load("model.pt").to(device)
此模型的子模块标识为
RecursiveScriptModule
。我想对上传的权重进行微调,以使其更简单,并将其转换为
torch.float32
最好将所有这些内容转换为普通PyTorch
nn.Module

在官方文件中,有人告诉我如何将
nn.Module
转换为
torchscript
,但我没有发现任何相反方向的例子。有办法做到这一点吗

p.S此处给出了加载模型和预训练模型的示例: