使用Pytorch训练,遇到数据类型与权重数据类型不匹配的解决方案:
Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.DoubleTensor) should be the same
- 将数据类型进行更改
# 将数据类型改为double,此data为Tensor数据
data.to(torch.double)
- 将权重(weight)类型进行更改
# 将模型权重改为FloatTensor,此model为模型
model.type(torch.FloatTensor)
以上两种方案都可以修改上述错误。