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

看板DataScience作者 (叫我松高魂 ~~)時間5年前 (2019/07/08 20:12), 編輯推噓1(109)
留言10則, 4人參與, 5年前最新討論串1/1
作業系統: 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

07/08 20:41, 5年前 , 1F
tf.keras.callbacks.ModelCheckpoint
07/08 20:41, 1F

07/08 20:46, 5年前 , 2F
tf.keras跟keras基本上是一樣的,除了backend放得位置
07/08 20:46, 2F

07/08 20:46, 5年前 , 3F
不一樣而已
07/08 20:46, 3F

07/08 21:59, 5年前 , 4F
s大想請教一下,您說的這個跟.save_model()在使用上有什麼
07/08 21:59, 4F

07/08 21:59, 5年前 , 5F
不同啊?如果我已經train完了,還可以使用checkpoint嗎?(
07/08 21:59, 5F

07/08 21:59, 5年前 , 6F
我看很多人都是在train的時候就放進callbacks參數裡面了)
07/08 21:59, 6F

07/08 23:13, 5年前 , 7F
train完就用save_model啊...,checkpoint是防止你
07/08 23:13, 7F

07/08 23:14, 5年前 , 8F
跑到一半程式掛掉又要重來
07/08 23:14, 8F

07/08 23:30, 5年前 , 9F
喔喔喔喔感謝感謝
07/08 23:30, 9F

07/11 08:52, 5年前 , 10F
跟掰陳一樣可以出去喊你在做AI了,參考看看
07/11 08:52, 10F
文章代碼(AID): #1T8pCKWv (DataScience)
文章代碼(AID): #1T8pCKWv (DataScience)