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