在实验pytorch跑实验时,遇到错误nll_loss_forward_reduce_cuda_kernel_2d_index” not implemented for ‘Int’,这是pytorch在计算交叉熵时出现的错误,一般在下面代码中出现
![](https://aitechtogether.com/wp-content/uploads/2023/09/4eaf4db4-b856-4b51-b509-b06768a57167.webp)
只要在loss这行的标签后面加上long()转化数据格式就可以。
文章出处登录后可见!
已经登录?立即刷新
在实验pytorch跑实验时,遇到错误nll_loss_forward_reduce_cuda_kernel_2d_index” not implemented for ‘Int’,这是pytorch在计算交叉熵时出现的错误,一般在下面代码中出现
只要在loss这行的标签后面加上long()转化数据格式就可以。
文章出处登录后可见!