会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
博客园
首页
新随笔
联系
订阅
管理
2024年2月4日
PyTorch下,使用list放置模块,导致计算设备不一的报错
摘要: 报错 在复现 Transformer 代码的训练阶段时,发生报错: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! 解决方案 通过
阅读全文
posted @ 2024-02-04 19:09 October-
阅读(49)
评论(0)
推荐(0)
编辑