Torch load model from gpu to cpu, so can convert to pytorch
安装load and savelua -i 开启交互式界面
require 导入包
‘torch’ : load
Sequential :’nn’ ‘cunn’ 根据提示的缺少来添加from gpu to cpu
m = torch.load(‘xxxx.t7’)
cudnn.conver(m,nn)
m = m:float()
torch.save(‘xxxx.cpu.t7’,m)
原创
2017-09-01 19:42:46 ·
1432 阅读 ·
0 评论