torch模型从GPU到CPU

最近在学习基于torch的文本识别。用的是VGG模型。

代码资源:https://github.com/bgshih/crnn

但是改模型是在GPU上训练的,我没有GPU所以要改到CPU上运行。这无疑是一个难题。在网上看到了一个改变的例子,借鉴过来参考。

原代码链接:https://github.com/vic-w/torch-practice/tree/master/mnist

阅读更多
个人分类: 深度学习
上一篇卷积神经网络(CNN)基础
下一篇中文汉字识别
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭