Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

使用pytorch测试模型时，报如下错误：

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

报错信息意思是说，我们输入的数据类型与网络参数的类型不符。

Input type的类型时torch.cuda.FloatTensor（GPU数据类型）

weight type（即net.parameters）为torch.FloatTensor（CPU数据类型）

解决方法如下：

• 使用以下语句将网络结构放入GPU中

model.cuda()

• 通过device()方法将模型放入GPU中

device = torch.device(‘cuda:0’)
net.to(device)

   该方法没有上一种便捷。

---

参考博客：https://www.pianshen.com/article/88881226547/