0

我试图将预先构建的 traced_torch_model.pt 加载到 C++;但是,错误消息显示“example-app.exe”存在代码 -1 请参见下面的我的代码片段和错误消息屏幕截图。感谢您的意见。 error_message_screenshot

using namespace std;
#include <torch/script.h>
#include <iostream>
#include <memory>

int main(int argc, const char* argv[]) {
    if (argc != 2) {
        std::cerr << "usage: example-app <C:\\Users\\C at work\\0813_2021\\traced_torch_model.pt\\>n";
            return -1;
    }

    torch::jit::script::Module module;
    try {
        // Deserialize the ScriptModule from a file using torch::jit::load().
        module = torch::jit::load(argv[1]);
    }
    catch (const c10::Error& e) {
        std::cerr << "error loading the model\n";
        return -1;
    }
    std::cout << "Model " << argv[1] << " loaded fine\n";

    // Create a vector of inputs.
    std::vector<torch::jit::IValue> inputs;
    inputs.push_back(torch::randn({ 1, 1, 64, 101 }));

    // Execute the model and turn its output into a tensor.
    at::Tensor output = module.forward(inputs).toTensor();
    std::cout << output << "\n";
    int y_hat = output.argmax(1).item().toInt();
    std::cout << "Predicted class: " << y_hat << "\n";
}
4

0 回答 0