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

标签：target   oss   范围   cross   rop   size   报错   batch   时报   

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

标签：target   oss   范围   cross   rop   size   报错   batch   时报   

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