Код: Выделить всё
module = torch::jit::load(model_path);Код: Выделить всё
mode_pathЯ проверил проблемы сохранения Python, и он был правильным и правильно сохраненным,
Я проверил путь к файлу .pt < /code>,
Я проверил, что DLL < /code> s находится в правильной палке, и все отсутствует, но все в порядке, и так, как мой cmake, file. class = "lang-cpp prettyprint-override">
Код: Выделить всё
int main()
{
torch::jit::script::Module module;
try
{
const std::string model_path = "simple_net.pt";
module = torch::jit::load(model_path);
torch::Tensor input = torch::rand({1, 4});
std::vector inputs;
inputs.push_back(input);
torch::Tensor output = module.forward(inputs).toTensor();
std::cout
Подробнее здесь: [url]https://stackoverflow.com/questions/79690376/pytorch-c-api-doesnt-load[/url]
Мобильная версия