首页 > 其他 > 详细

RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed

时间:2020-01-08 23:45:37      阅读:647      评论:0      收藏:0      [点我收藏+]

运行torch函数 torch.nn.functional.cross_entropy(pre, label, ignore_index=0)时报错,pre的shape为[ batch_size  , n]   label 的shape为[ batch_size].

其中batch_size是batch的大小,n为类别数

所以  label的每一个数的取值范围都应该在[ 0,n-1 ], 代表该下标的真实类别,即cur_target < n_classes‘

        label的每一个数的取值范围都应该大于0, 即cur_target >= 0 

RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed

原文:https://www.cnblogs.com/dongfangshenqi/p/12169027.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!