[問題] 已訓練好的模型如何儲存？

[sssh (叫我松高魂 ~~)]

作業系統: Win10

問題類別: 已訓練好的 CNN 模型儲存

使用工具: tensorflow

問題內容:

這幾天在看kaggle的某一個釋出kernel,跟著它的code跑到訓練結束，訓練花了四個小時跑，但是後面的code有點問題，我想再研究一下，請問我想先把整個訓練好的model儲存應該要怎麼做？我不想每次要開機都要重新訓練過啊啊啊啊啊啊.....

PS:kernel作者是用tf.keras，不是直接用 keras api

再麻煩各位前輩指點一下。

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 111.71.112.133 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/DataScience/M.1562587924.A.839.html
→ as30385438: tf.keras.callbacks.ModelCheckpoint
推 sxy67230: tf.keras跟keras基本上是一樣的，除了backend放得位置
→ sxy67230: 不一樣而已
→ sssh: s大想請教一下，您說的這個跟.save_model()在使用上有什麼
→ sssh: 不同啊？如果我已經train完了，還可以使用checkpoint嗎？（
→ sssh: 我看很多人都是在train的時候就放進callbacks參數裡面了）
→ as30385438: train完就用save_model啊...，checkpoint是防止你
→ as30385438: 跑到一半程式掛掉又要重來
→ sssh: 喔喔喔喔感謝感謝
→ erre: 跟掰陳一樣可以出去喊你在做AI了，參考看看