qq_43124103 2022-02-09 14:01 采纳率: 66.7%
浏览 474

关于libtorch,网络模型导入导出的问题,在c++中load报错,请问应该怎么检查,是什么原因呢?

在pycharm将训练好的网络模型加载号后,创建一个example输入,使用torch.jit.trace 保存的模型和torch.jit.script保存的模型都出现了torch::jit::Errorreport错误,是网路模型内部出现了其他不属于torch的变量导致的吗

c++代码
###
torch::jit::script::Module module;
//module = torch::jit::load("./resNetFpn-model-14_jit.pt");
module = torch::jit::load("./resNetFpn-model-14_jitscript.pt");###

运行结果及报错内容

img

  • 写回答

1条回答 默认 最新

报告相同问题?

问题事件

  • 创建了问题 2月9日