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

作者：小小林熬夜学编程 | 2024-02-28 16:09:47

踩

runtimeerror: expected backend cpu and dtype float but got backend cuda and

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/160043