逢甲大學

#新手 訓練完卻無法轉移訓練結果 (更 已解決 附上解決方案

7月27日 21:59
我參考這篇文章成功訓練出模型,可是我好像沒辦法把它的訓練結果保存下來 我用model.save_weight 保存用epoch=20 訓練後的權重 (訓練後的預測辨識率極高),開一個新的檔案重新建立一個模型,再把權重load進來後做預測,但一直沒辦法在新模型成功辨識,一定要再額外訓練幾次再預測才有效 想請問這是為什麼? 試著找資料可是又找不太到,也不太確定搜尋什麼關鍵字,卡了好幾天,求幫忙 😭 =====分隔線===== 更 抱歉!不是保存問題 😓😓 問題出在文章的最上方 characters 型態是 set ,set是無序的,導致每次執行時的英文對數字是不同的而無法正確得到結果 我的解決方法是 改成 list("0123...abc...ABC...") 把所有可能出現的字元都放進去 希望有同樣問題的人可以省下更多時間! 謝謝下方幫我思考過的各位!
2
回應 10
文章資訊
Logo
每週有 30 則貼文
共 10 則留言
國立成功大學
model.save_weight 只保存權重不保存模型
國立成功大學
原 PO - 逢甲大學
b1 所以…也要把模型保存下來的意思嗎? b2 可是我用同一個方式造模型出來(原文章裡的build_model) 各層名字應該是一樣的? 另外,可以跟你要這篇連結嗎? 🙏
國立成功大學
看完你的問題應該就解決了,存不存model整個架構由你決定,但如果要做fine tune 存權重就好
原 PO - 逢甲大學
b4 好,我研究一下,先謝謝你!
國立雲林科技大學
可以把模型架構存成一個檔案 用 main 檔去摳那個模型 再把全重載進去 save_weight 只存權重,沒存模型架構
原 PO - 逢甲大學
b4 我照著你給的那篇文章嘗試,裡面說的幾乎都試過了,但仍然無法解決我的問題…😅 想先確認一下,我只想保留訓練後的成果,只存權重是對的嗎? 我也試過用model.save 把整個模型+權重保存下來,再用load_model 載入,但仍然沒有保留先前的訓練成果… b6 是,你的方法我也試過,但仍是相同的問題 😅
匿名
此帳號疑似異常
官方正在進行身份確認
這則留言已被用戶檢舉含「交換個人資料」的內容。
原 PO - 逢甲大學
此問題已解決,更新在原文 怕之前有人看到這篇,仍在幫我嘗試,特此告知
國立成功大學
我之前是用字典存