keras/tensorflow 模型保存後重新加載準確率爲0 model.save and load giving different result

我在用別人的代碼跑程序的時候遇到了這個問題:
keras 模型保存後重新加載準確率爲0
GitHub上有個issue:model.save and load giving different result · Issue #4875 · keras-team/keras
這個issue有150個回覆了,大部分人都是遇到了這個問題。
我用tensorflow重寫了一遍,沒有解決問題。
然後我懷疑不是庫的問題,應該是其他部分代碼的問題。
最終我發現問題是在加載數據時使用了set去重,但是set是無序的,所以每次加載數據的標籤號都不一樣。把set轉換成list然後排序一下就解決了問題。
解決這個問題花了我整整2天。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章