解决pytorch当中RuntimeError: expected scalar type Double but found Float的问题

      出现这个问题的原因可能是因为tensor的数据类型不对,有可能是反向传播中输入x的类型不对,也有可能是训练和测试过程中的data类型不对,如果是反向传播过程的话,那就要看是哪一层神经网络出现问题,就在哪一层的前面加上x=x.to(torch.float32),比如我的是神经网络的第一层出现这个问题,就比如,我的是第一层卷积就出现这个问题,解决方法如下:

如果是在训练或者测试模型中出现这个问题,则解决方法如下:

 

 也有可能是训练或者测试过程中labels的类型不对,解决方法如下:

 


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