请选择 进入手机版 | 继续访问电脑版
查看: 1101|回复: 0

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

[复制链接]

665

主题

1234

帖子

6561

积分

xdtech

Rank: 5Rank: 5

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

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


回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

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