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

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

Сообщение Anonymous »

Я тестирую API CPP и пытаюсь загрузить файлы .pt < /code>, которые обучались на Python и сохранили там.module = torch::jit::load(model_path); (mode_path является строковой переменной и является Path to My .pt)
У меня нет видеокарты, и я использую только версию процессора обоих. проверил совместимость моего Pytorch в Python и CPP API,
Я проверил проблемы сохранения питона, и он был правильным и правильно сохраненным
Я проверил путь к файлу .pt < /code>,
я проверил, что dll < /code> s находится в правильной папке и есть все, но все в порядке, и так, как у меня, как cmake < /p, и все в порядке, и все в порядке, и так, как у меня, как я.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

Подробнее здесь: https://stackoverflow.com/questions/796 ... doest-load
Ответить

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

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

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

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

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