Pytorch C ++ API не загружаетсяC++

Программы на C++. Форум разработчиков
Ответить
Anonymous
 Pytorch C ++ API не загружается

Сообщение Anonymous »

Я тестирую API C ++ и пытаюсь загрузить файлы .pt < /code>, которые обучались на Python и сохранили там.

Код: Выделить всё

module = torch::jit::load(model_path);
(

Код: Выделить всё

mode_path
является строковой переменной и является Path to. Совместимость моего Pytorch в Python и C ++ API,
Я проверил проблемы сохранения 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]
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «C++»