![](https://img-blog.csdnimg.cn/20190918140145169.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
深度学习笔记
文章平均质量分 61
动手学深度学习笔记
新客草草
这个作者很懒,什么都没留下…
展开
-
Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
经过检查发现defforward内部定义了两个网络层,如上图红圈处,将这两个层定义到__init__()中,然后在forward调用这两个层,修改后如下图。注意任何标题提示的报错都可以用下面方法的原理找出问题数据(在CPU),然后将该数据放入GPU即可。在运行bert模型时,打算将代码放到服务器的GPU上面运行,将模型和设置的参数。通过检查自己定义的所有类,发现这个类有问题。数据从cpu到gpu的迁移如下。...原创 2022-07-27 09:59:11 · 2309 阅读 · 0 评论 -
Lenet手写数字识别代码解析
Lenet的MxNet实现代码逐句分解原创 2021-12-10 10:29:57 · 966 阅读 · 0 评论