出现这个错误一般是由于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()