本文主要是介绍6、RuntimeError: “log_softmax_lastdim_kernel_impl“ not implemented for ‘Long‘,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
问题:
在训练模型,计算损失函数的时候,报如下错误:
解决方法:
1、查看预测值和标签的数据类型,使用dtype方法;
2、将预测值的数据类型改为torch.float32;
这篇关于6、RuntimeError: “log_softmax_lastdim_kernel_impl“ not implemented for ‘Long‘的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!