作者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