1.調試程序發現計算loss的時候出現如題所示錯誤，導致后續程序報CUDA的錯
nn.CrossEntropyLoss()，我設置ignore_index為0，mask設為0,1,2，做2分類，就會這樣
2.將ignore_index改為-1，計算loss的時候把mask=mask-1，就能正常分類
不知道為啥會出現這種情況
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
torch.Tensor和torch.tensor的區別
torch.tensor(),torch.Tensor()
torch.Tensor.view (Python method, in torch.Tensor)
torch.tensor拼接與list(tensors)
torch.Tensor和numpy.ndarray
torch.Tensor中的clone()方法
torch.Tensor文檔學習筆記
torch.tensor拼接與list(tensors)
torch.Tensor 與 numpy.ndarray的相互轉化
torch.tensor默認requires_grad=False