报错:
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
解决方案:
最近跑代码,在训练时出现标题所示问题。浏览了很多方法后,总结出现这个问题的主要原因是输入的数据类型与网络参数的类型不符。
解决方案 加一行model.cuda()
model=newmodel.Model()
model.cuda()
报错:
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
解决方案:
最近跑代码,在训练时出现标题所示问题。浏览了很多方法后,总结出现这个问题的主要原因是输入的数据类型与网络参数的类型不符。
解决方案 加一行model.cuda()
model=newmodel.Model()
model.cuda()