1 问题描述
Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
输入input和参数weight类型应该一致。
2 分析
经过查找后,发现在读取图片和标签的时候,并没有对其类型进行转化,就可能根据图像本身的大小设置,代码中出现的类型刚开始是float32,后边有的图像是float64,导致输入与参数类型不匹配。
3 结论
读取图像和标签时,应该人为设定其类型,将所有的数据转化为同一类型。
参考文献:【已解决】Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
关于标签要设置为long类型:【pytorch】【已解决】RuntimeError: Expected object of scalar type Long but got scalar type Int for argument
关于图像设置为float32或者64