作者sssh (叫我松高魂 ~~)
看板DataScience
標題[問題] 已訓練好的模型如何儲存？
時間Mon Jul 8 20:12:01 2019
作業系統: Win10
問題類別: 已訓練好的
CNN 模型儲存
使用工具: tensorflow
問題內容:
這幾天在看kaggle的某一個釋出kernel,跟著它的code跑到訓練結束，訓練花了四個小時跑，但是後面的code有點問題，我想再研究一下，請問我想先把整個訓練好的model儲存應該要怎麼做？我不想每次要開機都要重新訓練過啊啊啊啊啊啊.....
PS:kernel作者是用tf.keras，不是直接用 keras api
再麻煩各位前輩指點一下。
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 111.71.112.133 (臺灣)
※ 文章網址: https://webptt.com/m.aspx?n=bbs/DataScience/M.1562587924.A.839.html
1F：→ as30385438: tf.keras.callbacks.ModelCheckpoint 07/08 20:41
2F：推 sxy67230: tf.keras跟keras基本上是一樣的，除了backend放得位置 07/08 20:46
3F：→ sxy67230: 不一樣而已 07/08 20:46
4F：→ sssh: s大想請教一下，您說的這個跟.save_model()在使用上有什麼 07/08 21:59
5F：→ sssh: 不同啊？如果我已經train完了，還可以使用checkpoint嗎？（ 07/08 21:59
6F：→ sssh: 我看很多人都是在train的時候就放進callbacks參數裡面了） 07/08 21:59
7F：→ as30385438: train完就用save_model啊...，checkpoint是防止你 07/08 23:13
8F：→ as30385438: 跑到一半程式掛掉又要重來 07/08 23:14
9F：→ sssh: 喔喔喔喔感謝感謝 07/08 23:30
10F：→ erre: 跟掰陳一樣可以出去喊你在做AI了，參考看看 07/11 08:52