在实验pytorch跑实验时,遇到错误nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int',这是pytorch在计算交叉熵时出现的错误,一般在下面代码中出现文章来源:https://www.toymoban.com/news/detail-506694.html
只要在loss这行的标签后面加上long()转化数据格式就可以。文章来源地址https://www.toymoban.com/news/detail-506694.html
到了这里,关于“nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘ 问题解决的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!