當(dāng)前位置：首頁(yè)
> 深度學(xué)習(xí)損失函數(shù)選擇
解決 nll_loss_forward_reduce_cuda_kernel_2d_index not implemented for 'float' 錯(cuò)誤的有效方法
1個(gè)月前 (05-13)CN2資訊
面對(duì) PyTorch 中的常見(jiàn)錯(cuò)誤 'nll_loss_forward_reduce_cuda_kernel_2d_index not implemented for 'float''，本文探討了 NLL Loss 的工作原理及解決方案。通過(guò)正確的數(shù)據(jù)類(lèi)型選擇，切換數(shù)據(jù)格式，甚至自定義 CUDA 內(nèi)...