训练模型后,我使用 Pytorch 1.5 在 Python 端创建了一个跟踪文件。我尝试使用 libtorch 1.5 编写该代码在 C++ 端加载该文件:
int main(int argc, const char* argv[]) {
//Deserialize the ScriptModule from a file using torch::jit::load().
torch::jit::script::Module module = torch::jit::load("D:/Downloads/deneme18.pt");
module.to(torch::kCUDA);
}
它给了我这个错误信息:
“DL_Predicton.exe 中 0x00007FF8C2A0A799 处未处理的异常:Microsoft C++ 异常:内存位置 0x0000009B6118EFE0 处的 c10::Error。”