自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(2)
  • 收藏
  • 关注

原创 Ptorch 与libTorch 使用过程中问题记录

最近在C++调用pytorch模型的时候遇到了一点小问题,特此记录。基于python和pytorch训练了一个分类模型,保存为pkl文件;因C++和Python读取的文件方式不同,所以在使用C++调用之间把训练生成的模型文件转换成Torch Script,即生成一个pt文件,这样就可以使用C++进行调用。生成的pkl和pt文件在python下测试,结果保持一致,证明模型没有问题,很好,继续。...

2019-04-23 15:46:36 2946

原创 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 11593 13

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除