“load”: 不是“torch::jit”的成员

举报
风吹稻花香 发表于 2021/06/05 23:57:59 2021/06/05
【摘要】   严重性    代码    说明    项目    文件    行    禁止显示状态 错误    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.

 


  
  1. #include <torch/script.h>
  2. torch::DeviceType device_type;
  3. if (torch::cuda::is_available()) {
  4. device_type = torch::kCUDA;
  5. }
  6. else {
  7. device_type = torch::kCPU;
  8. }
  9. torch::Device device(device_type);
  10. // Deserialize the ScriptModule from a file using torch::jit::load().
  11. //反序列化pytorch模型,并加载到module变量
  12. std::shared_ptr<torch::jit::script::Module> module = torch::jit::load(argv[1], device);
  13. assert(module != nullptr);

torch::jit 没有成员load

 

文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/jacke121/article/details/93322398

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。