“load”: 不是“torch::jit”的成员
【摘要】
严重性 代码 说明 项目 文件 行 禁止显示状态 错误 C2039 “lo...
严重性 代码 说明 项目 文件 行 禁止显示状态
错误 C2039 “load”: 不是“torch::jit”的成员 simnet d:\soft\libtorch-win-shared-with-deps-latest\build_person\main2.cpp 59
解决:
加上头文件:
#include <torch/script.h> // One-stop header.
-
#include <torch/script.h>
-
-
torch::DeviceType device_type;
-
-
if (torch::cuda::is_available()) {
-
device_type = torch::kCUDA;
-
}
-
else {
-
device_type = torch::kCPU;
-
}
-
torch::Device device(device_type);
-
-
-
// Deserialize the ScriptModule from a file using torch::jit::load().
-
//反序列化pytorch模型,并加载到module变量
-
std::shared_ptr<torch::jit::script::Module> module = torch::jit::load(argv[1], device);
-
-
assert(module != nullptr);
torch::jit 没有成员load
文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。
原文链接:blog.csdn.net/jacke121/article/details/93322398
【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)