pytorch tensor int型除法

昨天晚上跑起來一個classification實驗,今天發現訓練loss在降,然而accuracy永遠是0 。。。直覺告訴我evaluation有問題 然後發現自己寫了個很愚蠢的bug accuracy對應的tensor出來是int型的,我用到了一個除法取平均。而pytorch裏無論用 / or // 結果都是取整,,accuracy一直就沒有。。 所以轉換成float就沒問題了,,低級bug,下
相關文章
相關標籤/搜索