expected scalar type Long but found XXXXX

Posted 2023-03-01 GaoZhenwen2

初学pytorch的报错 记录一下啊。

网络结构

搭建的是一个mnist数据集的分类神经网络。

损失函数是用的交叉熵损失函数

loss_func = F.cross_entropy

数据集

mnist数据集用的是离线下载下来的csv文件，只用pandas做了简单的处理，然后转成了tensor,用搭建的网络开始训练，然后训练的时候开始报错了。

错误类型

expected scalar type Long but found XXXXX

解决方法

转成tensor的时候数据类型改成如下:

输入数据使用torch.float32

标签数据使用torch.int64

x_train ,y_train,x_vaild,y_vaild = x_train.type(torch.float32) ,y_train.type(torch.int64),x_vaild.type(torch.float32),y_vaild.type(torch.int64) 

总结

pytorch学的不够系统，只是跟着网上的视频学了个框架，数据类型都对应不上，还是要看书啊。