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

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

猜你喜欢

转载自www.cnblogs.com/dongfangshenqi/p/12169027.html