解决“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()转化数据格式就可以。

物联沃分享整理
物联沃-IOTWORD物联网 » 解决“nll_loss_forward_reduce_cuda_kernel_2d_index“ not implemented for ‘Int‘问题

发表回复