libtorch新的版本如何调用.pt文件
一、在pytorch1.2版本中,我们使用Visual Studio时调用.pt文件时,会报错,如下
1、不存在用户定义的从"torch::jit::script::Module"到std::shared_ptr torch::jit::script::Module
的适当转换。
二、解决
1、旧版本1.0调用
std::shared_ptr<script::Module>
2、Version1.1
using torch::jit::script::Module;
std::shared_ptr<Module> m=torch::jit->forward(...);
3、Version 1.2
using torch::jit::script::Module;
Module m=torch::jit::load("D:\\ResNet_jit.pt");
m.forward(...);
三、详见github官网
https://github.com/pytorch/pytorch/releases/tag/v1.2.0