unable to get repr for class ‘torch.tensor’
1.調試程序發現計算loss的時候出現如題所示錯誤,導致后續程序報CUDA的錯 nn.CrossEntropyLoss(),我設置ignore_index為0,mask設為0,1,2,做2分類,就會這樣 2.將ignore_index改為-1,計算loss的時候把mask=mask-1 ...
1.調試程序發現計算loss的時候出現如題所示錯誤,導致后續程序報CUDA的錯 nn.CrossEntropyLoss(),我設置ignore_index為0,mask設為0,1,2,做2分類,就會這樣 2.將ignore_index改為-1,計算loss的時候把mask=mask-1 ...