先看另一位博主的文章:训练和测试数据转换不对的修改方法
如果你找到的不是因为data或者label没有加上
.to(device)
的话,
而是因为调用了torchmetrics的评价指标如Accuracy,F1等的话,那么就要注意是不是因为这些指标和数据不在同一块设备上。
尝试将
from torchmetrics import Accuracy
acc = Accuracy(number_classes=2) # 这里的参数不一定一致
改成
from torchmetrics import Accuracy
acc = Accuracy(number_classes=2).to(device)
问题解决