问题描述:
按照PyTorch中文教程的【 在 C++ 中加载 PYTORCH 模型 】一文,尝试调用 PyTorch模型。
1. 例子来源
在 C++ 中加载 PYTORCH 模型
我是使用Qt新建纯C++工程,然后CMake编译、运行的,直接贴代码:
1.1 C++代码
main.cpp
#include <torch/script.h> // One-stop header.
#include <torch/torch.h>
#include <iostream>
#include <memory>
using namespace std;
int main()
{
// std::shared_ptr<torch::jit::script::Module> module = torch::jit::load(
// "/home/book/Jupyterlab work/model.pt");
torch::jit::script::Module module = torch::jit::load(
"/home/book/Jupyterlab work/model.pt");
// C