在實驗pytorch跑實驗時,遇到錯誤nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int',這是pytorch在計算交叉熵時出現(xiàn)的錯誤,一般在下面代碼中出現(xiàn)文章來源:http://www.zghlxwxcb.cn/news/detail-506694.html


只要在loss這行的標(biāo)簽后面加上long()轉(zhuǎn)化數(shù)據(jù)格式就可以。文章來源地址http://www.zghlxwxcb.cn/news/detail-506694.html
到了這里,關(guān)于“nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘ 問題解決的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!