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 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM