我的一个解决方法是降低torch的版本
出错误的版本:torch 1.13.0+cu117
不出错误的版本:torch 1.1.1+cu113
indices should be either on cpu or on the same device as the indexed tensor (cpu)
最新推荐文章于 2024-07-27 12:20:46 发布
我的一个解决方法是降低torch的版本
出错误的版本:torch 1.13.0+cu117
不出错误的版本:torch 1.1.1+cu113