这个问题真的是非常烦人,在自己目前的项目里经常出现,而且报错位置不止一处,非常没有规律。看了网上很多经验贴,最后自己是这么解决的:
所有关于模块(包括网络架构、数据集的数据和标签、损失函数)转移到CUDA上的语句从.to(device)
全部改成.cuda()
,就奇迹般的好了
这个问题真的是非常烦人,在自己目前的项目里经常出现,而且报错位置不止一处,非常没有规律。看了网上很多经验贴,最后自己是这么解决的:
所有关于模块(包括网络架构、数据集的数据和标签、损失函数)转移到CUDA上的语句从.to(device)
全部改成.cuda()
,就奇迹般的好了