pytorch挖坑之:Expected object of scalar type Long but got scalar type Float for argument target...

目錄 1. 問題分析 2. 總結 1. 問題分析 報錯提示:需要Long類型的,但是收到了Float類型。 google了一下,都說是到相應的位置,後面加上.float()。 但是改了之後,還是不行,報了另外一個錯誤: 後來發現,應該是自己傳的target有問題,也就是label。 查看了一下自己寫的Dataset類,手動地把0變成[1, 0], 把1變成[0, 1], **其實根本不需要!!!不
相關文章
相關標籤/搜索