RuntimeError: expected backend CPU and dtype Float but got backend CUDA and dtype Float

最新推荐文章于 2023-08-07 09:37:54 发布

graph builder

最新推荐文章于 2023-08-07 09:37:54 发布

阅读量2.1k

点赞数

分类专栏： AI for CV

本文链接：https://blog.csdn.net/qq_36783816/article/details/102624800

版权

AI for CV 专栏收录该内容

20 篇文章 0 订阅

订阅专栏

                    
                    方法：数据没有放进gpu造成，一般加.cuda()可以解决
原因：在加载优化器和模型的训练参数时，没有注意到代码的执行顺序，应该在加载模型参数后，就要将模型放到gpu上，然后再加载优化器参数；但是我的执行代码却是所有参数加载好后，将模型放到gpu上，这样就出现了优化器参数分离；优化器加载的参数在cpu上，但是优化器在使用时，却是在gpu上。也就是数据没有放到gpu上。