【已解决】【读取图像以及标签应该人为设置类型】Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)

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