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