设为首页收藏本站
切换到宽版
易学智能»易学智能论坛 › 技术大讨论 › PyTorch › 常见错误：RuntimeError: expected type torch.FloatTen ...
返回列表 发新帖
查看: 1320|回复: 0

常见错误：RuntimeError: expected type torch.FloatTensor but got torch.cu...

[复制链接]
shaoheshaohe

665

主题

1234

帖子

6670

积分

xdtech

Rank: 5Rank: 5

积分
6670
发表于 2020-5-13 08:36:10 | 显示全部楼层 |阅读模式
计算中有的参数为cuda型有的参数却是cpu型就会遇到这样的错误。

注意：torch model加载，要先将模型to cuda设备，再加载weights。不然，梯度传播会遇到以上错误


回复

使用道具 举报

返回列表 发新帖
高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表