python报错：(torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

原因：这是因为您的模型权重 (weight) 和输入数据 (input) 不在同一个设备上（一个在 GPU 上，另一个在 CPU 上）

解决：

一、在模型前面加代码：

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

二、然后，在调用模型到cpu上

原版代码：

model = DiCNN().cpu()

改成：

model = DiCNN().to(device)