6、RuntimeError: “log_softmax_lastdim_kernel_impl“ not implemented for ‘Long‘

问题:

        在训练模型,计算损失函数的时候,报如下错误:

解决方法:

        1、查看预测值和标签的数据类型,使用dtype方法;

        2、将预测值的数据类型改为torch.float32;

 

 


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部