在实验pytorch跑实验时，遇到错误nll_loss_forward_reduce_cuda_kernel_2d_index” not implemented for ‘Int’，这是pytorch在计算交叉熵时出现的错误，一般在下面代码中出现
只要在loss这行的标签后面加上long()转化数据格式就可以。
文章出处登录后可见！
已经登录？立即刷新
“nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘ 问题解决
赞 (0)
虚拟环境下更换python版本
上一篇
2023年9月7日