
pytorch研究与应用
seasermy
专注AI算法研究与应用,AI软硬件协同优化,AI系统解决方案
展开
-
RCAN代码问题】Pytorch1.0版本无法引用‘_worker_manager_loop’模块
在调试超分辨率RCAN模型的Pytorch工程时,遇到问题:ImportError: cannot import name '_worker_manager_loop' from 'torch.utils.data.dataloader'查看dataloder源码发现问题原因是Pytorch升级到1.0后,dataloder模块中的_worker_manager_loop函数已被移除并由...转载 2019-07-02 18:11:14 · 1749 阅读 · 0 评论 -
注意一下cv::Mat的roi及clone
今天在执行libtorch程序的时候,有如下两句:cv::Mat img_patch = img_float(cv::Range(pos[0], pos[1]), cv::Range(pos[2], pos[3])); // img_patch为img_float的sub imageat::Tensor image_tensor = torch::from_blob(img_patch....原创 2019-08-14 17:51:20 · 1584 阅读 · 2 评论 -
pytorch c++部署编译?libtorch库加载报错?
pytorch有几个优点:(1)强大的python接口,便于实验,比起tensorflow方便(2)动态图计算,写网络就像写普通的加减乘除表达式一样,灵活组建网络,便于调试,mxnet等frame work也借鉴了这种模式,在这鄙视一下tensorflow,久久不支持动态图,直到去年2.0才效仿了动态图,美其名曰eager execution.pytorch的c++接口也发布了(...原创 2019-08-12 14:34:02 · 5973 阅读 · 7 评论 -
如何在c++平台上部署pytorch模型?
pytorch接口简单灵活,深受深度学习研究者的喜爱,不少论文及github上的开源代码是用pytorch写的,那么,训练完pytorch模型后,部署到c++平台上,着实让不少人头疼.好在,pytorch开放了libtorch c++接口,至此,caffe, mxnet, tensorflow, pytorch均完善了python以及c++接口,无论是在PC端,还是在移动端,均可满足大多数人的训练...原创 2019-08-16 15:40:19 · 11669 阅读 · 1 评论