在訓練的train.py改爲：
running_corrects += float(torch.sum(preds == labels.data))
是不是pytorch的版本問題我不知道，主要原因還是因爲sum中的數值如果是bool類型，那麼輸出的就只有０和１，所以要將sum中數據類型強行轉換，轉換成float的類型，就會解決這種情況
Accuracy一直顯示０
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章
在訓練的train.py改爲：
running_corrects += float(torch.sum(preds == labels.data))
是不是pytorch的版本問題我不知道，主要原因還是因爲sum中的數值如果是bool類型，那麼輸出的就只有０和１，所以要將sum中數據類型強行轉換，轉換成float的類型，就會解決這種情況