花費 6 ms
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 ...

Sat Jun 19 11:47:00 CST 2021 0 347

 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM