“nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘ 问题解决

在实验pytorch跑实验时,遇到错误nll_loss_forward_reduce_cuda_kernel_2d_index” not implemented for ‘Int’,这是pytorch在计算交叉熵时出现的错误,一般在下面代码中出现

只要在loss这行的标签后面加上long()转化数据格式就可以。

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2023年9月7日
下一篇 2023年9月7日

相关推荐

此站出售,如需请站内私信或者邮箱!