上述为模型中部分自定义模块的python、C++实现,如何使用libtorch部署模型,如何才能让C++正确识别python导出的.pt模型中这些自定义模块呢?请教一下具体的流程
![](https://profile-avatar.csdnimg.cn/1843da2c71cd43f995d54121b5c40a1d_qq_44895181.jpg!4)
libtorch模型部署
上述为模型中部分自定义模块的python、C++实现,如何使用libtorch部署模型,如何才能让C++正确识别python导出的.pt模型中这些自定义模块呢?请教一下具体的流程
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
- 邀请回答
-
4条回答 默认 最新
- 爱晚乏客游 2023-06-25 10:24关注
你保存模型的时候把网络结构一起保存下来无法读取吗?按理来说直接带网络结构就不需要在cpp下面定义网络结构的。
另外,cpp部署的话很少直接读取pt或者pth文件,都是转到onnx或者TorchScript,按照你的libtorch方式部署,我更建议你转成torch.jit.ScriptModule
https://zhuanlan.zhihu.com/p/486914187解决 无用评论 打赏 举报 编辑记录
悬赏问题
- ¥15 mmo能不能做客户端怪物
- ¥15 osm下载到arcgis出错
- ¥15 Dell g15 每次打开eiq portal后3分钟内自动退出
- ¥200 使用python编写程序,采用socket方式获取网页实时刷新的数据,能定时print()出来就行。
- ¥15 matlab如何根据图片中的公式绘制e和v的曲线图
- ¥15 我想用Python(Django)+Vue搭建一个用户登录界面,但是在运行npm run serve时报错了如何解决?
- ¥15 QQ邮箱过期怎么恢复?
- ¥15 登录他人的vue项目显示服务器错误
- ¥15 (标签-android|关键词-app)
- ¥15 comsol仿真压阻传感器