【无标题”nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int“,Python运行程序时出现的问题

错误原因:
情况一:数据类型或者Label的类型错误,需要转换
情况二:分类识别问题中,标签被需要转换为one—hot编码形式

解决方案:
针对情况一,可直接将源代码中的
在这里插入图片描述

转换为:
在这里插入图片描述

针对情况二,可以利用函数scatter函数,详见官方说明https://blog.csdn.net/duan_zhihua/article/details/82556676

你可能感兴趣的:(python,opencv,机器学习,计算机视觉,深度学习)