Pytorch数据类型与模型权重不匹配

使用Pytorch训练，遇到数据类型与权重数据类型不匹配的解决方案：

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

1. 将数据类型进行更改

# 将数据类型改为double，此data为Tensor数据
data.to(torch.double)

2. 将权重(weight)类型进行更改

# 将模型权重改为FloatTensor，此model为模型
model.type(torch.FloatTensor)

以上两种方案都可以修改上述错误。

标签：匹配,weight,权重,FloatTensor,torch,数据类型,Pytorch,type
来源： https://blog.csdn.net/weixin_41032661/article/details/121705849