Grandret的博客

阅读官网文档发现pytorch除了使用torch.save与torch.load保存加载数据/模型外还可以使用torch.jit.save(),torch.jit.load()。 后者的好处在于 其他py脚本在使用模型时不需要将原模型所在py文档也引入 可以跨平台使用 官网代码 scripted_module = torch.jit.script(MyModule()) torch.jit.save(scripted_module, 'mymodule.pt') torch.jit.load('mymo