/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:106: void cunn_ClassNLLCriterion_updateOutput_kernel(D

50 篇文章 6 订阅
12 篇文章 4 订阅

welcome to my blog

头一次使用pytorch, 调用交叉熵损失函数nn.CrossEntropyLoss()的时候报错
/pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:106: void cunn_ClassNLLCriterion_updateOutput_kernel(Dtype *, Dtype *, Dtype *, long *, Dtype *, int, int, int, int, long) [with Dtype = float, Acctype = float]: block: [0,0,0], thread: [4,0,0] Assertion t >= 0 && t < n_classes failed.

原因: nn.CrossEntropyLoss(input, target)中, input的列数和target不匹配

举例来说, 如果是个5分类问题, 那么input的形状是batch_size * 5, target中每个元素的取值范围是{0,1,2,3,4}, 也就是[0, 类别数量减1]
所以对于n分类问题来说, 如果target中某个元素的取值不在[0,n-1]中就会报上述错误

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值