數據出現None的現象因爲函數沒有調用進去
重新查詢相應類,有沒有返回值return
並且要注意模型是否使用cuda
數據出現大量的Nan的情況,其他數據已經十億開外,明明每一步的張量形狀都是正確的,但是就是出現各種索引錯誤,無法進行正確的張量形變工作。 經過師弟幫忙,一部一部進行測試,部分排除錯誤問題。 使用 torch.randn(8,3,2
被自己都蠢哭了~~ 一直報錯沒有調整過來,以爲是key的問題,後來發現是自己導入key的方式有問題,週末要好好研究一下這個 state_dict,以免之後再次出現這種超級低級的錯誤!! 之前的問題是 unexpected key
在訓練的train.py改爲: running_corrects += float(torch.sum(preds == labels.data)) 是不是pytorch的版本問題我不知道,主要原因還是因爲sum中的數值如果是b