一些关于Torch7的记录

本文记录了在Torch7中将模型从GPU迁移到CPU时遇到的两种常见错误:读取模型错误和nil类型错误。针对这些问题,提供了解决方案,包括检查导入的包、确定模型类型以及在lua table中查找和提取模型。通过这些步骤,可以成功地在无GPU环境中加载和转换模型。
摘要由CSDN通过智能技术生成

有时想要在一台有GPU的机器上(前提是这台机器已经装好了CUDA)训练好模型之后将模型转到CPU型的,这样就可以在没有GPU的机器上(或者没装cuda加速的机器)导入这个模型了。但是可能会遇到奇怪的错误,这里就记录一些贫僧遇到的奇怪的错误。

奇怪错误之读取模型

在进入了torch的交互命令行环境(就是用th来进入的那个环境)之后,如果发现用m = torch.load('x.t7')遇到了这种unknown Torch class <nn.gModule>
stack traceback:
[C]: in function 'error'
错误的话,那么很有可能没有导入全要导入的包,例如这里就少导入了require 'nngraph'包,导入之后就可以读取模型了。

奇怪错误之nil什么什么的

使用m = torch.load('x.t7')来读取模型(t7文件,鬼知道里面存了什么。。。)成功后,使用m = m:float()来转化模型的时候,如果遇到了这个错误:

attempt to call method 'float
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值