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

原因是没有把输入数据类型转为float类型

首先找到代码出错的位置,将该处的数据类型转为float类型:

x = x.type(torch.FloatTensor)

 


版权声明:本文为weixin_38314865原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。