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)
  1. 将权重(weight)类型进行更改
# 将模型权重改为FloatTensor,此model为模型
model.type(torch.FloatTensor)

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

上一篇:elfinder源码浏览-Volume文件系统操作类(1)


下一篇:Python面向对象(类的继承与多态)