- 博客(2)
- 收藏
- 关注
RSS订阅原创 Ptorch 与libTorch 使用过程中问题记录
最近在C++调用pytorch模型的时候遇到了一点小问题，特此记录。基于python和pytorch训练了一个分类模型，保存为pkl文件；因C++和Python读取的文件方式不同，所以在使用C++调用之间把训练生成的模型文件转换成Torch Script，即生成一个pt文件，这样就可以使用C++进行调用。生成的pkl和pt文件在python下测试，结果保持一致，证明模型没有问题，很好，继续。...
2019-04-23 15:46:36
2982
原创 Pytorch C++部署 之 TorchScript 踩坑记录
小白一枚，最近尝试用c++调pytorch模型，网上搜了一对资料，大同小异，都是需要先将模型文件转换成torch script。pytorch官网提供两种方式，1：trace 2：注释。这两种方式怎么操作，官网以及很多博客都有详细介绍，不细扣，我也扣不明白，毕竟我是个小白，放链接供参考：https://blog.csdn.net/dou3516/article/details/829124...
2019-04-04 10:41:31
11844
13
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人