作者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/cn.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