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

出现这个错误一般是由于cuda和cpu格式没匹配

  • 输入的数据类型为torch.cuda.HalfTensor,说明输入数据在GPU中
  • 模型参数的数据类型为torch.FloatTensor,说明模型还在CPU

加载到cuda:

model.cuda() 或

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

如果是数据没有载到gpu:

image=image.cuda()

你可能感兴趣的:(python,深度学习,人工智能)