Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

1779-孙同学

发表文章数:101

首页 » 算法 » 正文

常规解决方案

从报错问题描述中可以找到错误原因

输入的数据类型为torch.cuda.FloatTensor,说明输入数据在GPU中
模型参数的数据类型为torch.FloatTensor,说明模型还在CPU
问题原因搞清楚了,模型没加载到CPU,在代码中加一行语句就可以了

model = model.cuda()
model = model.to('cuda')
model.cuda()
model.to('cuda')

上面四行任选一

问题是加在哪里

一般在summary模型的时候提示错误, 所以可以加到summary之前
原代码:

summary(chartextcnn, input_size=(70,1014))
# tensorflow: bn: 256*4

改之后:

chartextcnn = chartextcnn.to('cuda')
summary(chartextcnn, input_size=(70,1014))
# tensorflow: bn: 256*4

未经允许不得转载:作者:1779-孙同学, 转载或复制请以 超链接形式 并注明出处 拜师资源博客
原文地址:《Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same》 发布于2021-10-13

分享到:
赞(0) 打赏

评论 抢沙发

评论前必须登录!

  注册



长按图片转发给朋友

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

Vieu3.3主题
专业打造轻量级个人企业风格博客主题!专注于前端开发,全站响应式布局自适应模板。

登录

忘记密码 ?

您也可以使用第三方帐号快捷登录

Q Q 登 录
微 博 登 录