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‘_第1张图片

6、RuntimeError: “log_softmax_lastdim_kernel_impl“ not implemented for ‘Long‘_第2张图片

 

 

你可能感兴趣的:(杂项,深度学习,人工智能)