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

别等时光非礼了梦想. 提交于 2020-01-08 23:13:07

运行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 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!