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

错误报在了forward里的Conv2d处。原因是函数写在forward里可能默认cpu，如果写在init构造函数里，就不需要再指定cuda。

修改为箭头指示就不再报错了。

 

【参考】

Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same-CSDN博客

posted @   夕西行  阅读(98)  评论(0)  编辑  收藏  举报
相关博文：
· 深度学习各种错误汇总
· moveToThread()、QtConcurrent注意项
· RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
· Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same(解决)
· [报错]-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same
阅读排行：
· DeepSeek 开源周回顾「GitHub 热点速览」
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布：重大改进与新特性概览！
· AI与.NET技术实操系列（二）：开始使用ML.NET
· 单线程的Redis速度为什么快？
历史上的今天：
2019-12-14 指针的引用
2019-12-14 初始化、赋值
2019-12-14 优质学习资料总结
2019-12-14 对象的生命历程
2019-12-14 构造函数、拷贝构造函数、析构函数
点击右上角即可分享
微信分享提示
AI FOR CODE 大赛