RuntimeError: Expected object of backend CPU but got backend CUDA for argument #3 ‘index‘

问题: 后端 Cuda 的预期对象, 但得到了后端 Cpu 的论点#3 "指数" 

解决办法:未能将张力转移到 GPU。

尝试model.to(device)

猜你喜欢

转载自blog.csdn.net/weixin_45223645/article/details/120288670
今日推荐