狗bug，弄了我一天。。。
原因是在用GPU跑代码的时候，网络模型中使用了多个类，结果习惯性地只给主类使用了.to(device)，网络中的其他类没管了，导致出现这个bug，后来在涉及到的类后面都添加了.to(device)后就OK了
block = Block().to(device) # 次类
net = Uet2plus().to(device) # 网络中的主要类
| |
|
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
| -> Python知识库 -> RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the -> 正文阅读 |
|
|
[Python知识库]RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the |
|
|
|
|
|
|
| 上一篇文章 下一篇文章 查看所有文章 |
|
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
| 360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 | -2024/11/15 11:28:20- |
|
| 网站联系: qq:121756557 email:121756557@qq.com IT数码 |