黑马程序员技术交流社区

标题: 【上海校区】tensorflow的模型保存于读取 [打印本页]

作者: 不二晨 时间: 2018-11-19 10:03
标题: 【上海校区】tensorflow的模型保存于读取
在模型的保存部分有两个重要的函数，

第一个  :  saver = tf.train.Saver(  )

我的理解是：这行代码大概就相当于实例化；

这个函数有一个很重要的参数，max_to_keep，这个值决定了你要保存最后几次的模型

可能这样说很难让人理解，其实就是这样；

max_to_keep=1  我们保存最后那次修炼的模型  ( 譬如说我们的 epoch = 10, 那么就只会保存第十次训练的模型)

max_to_keep=10 . 则这10次的模型就都会被保存下来，（然而个人感觉这并没有什么卵用）

现在我们有了一个实例化的object了，我们来用第二个函数

saver.save(sess,'ckpt/mnist.ckpt',global_step=step)
从名字就不难看出来，这行代码的作用就是保存，

第一个参数sess自然就不必多说了，第二个参数设定保存的路径和名字（保存在ckpt文件夹下，名字为mnist. ckpt）

第三个参数严格意义上来说也是用来命名的，他们将训练的次数作为后缀加入到你的文件名字中

譬如说我们的epoch = 10（我的意思是step = 10）

那么我们执行上面的语句，你ckpt文件夹下的文件就会是这样：  mnist-10.ckpt

因为不是科班出身，我并没有学过数据库，所以对pointcheck的定义也很含糊，我争取这周简单的看一下这方面的知识然后在回来充分的理解一下。。。。。

在附上我看过的觉得挺好的一篇博客，和大家分享一下https://blog.csdn.net/Laurenitum0716/article/details/79326128

-  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -  -

经过上面的两步，我们的模型就保存完毕了，

如果我们想要恢复这个模型，则要用到是restore()函数

我们看到，这个restore一共3个参数，self,  sess  自然不必说

第三个参数： save_path指的是保存的模型路径。

我们可以使用tf.train.latest_checkpoint（）来自动获取最后一次保存的模型。

---------------------
【转载】
作者：Pierce_KK
原文：https://blog.csdn.net/Pierce_KK/article/details/84073971

作者: 梦缠绕的时候 时间: 2018-11-20 15:53

作者: 小影姐姐 时间: 2018-11-21 16:33

作者: 不二晨 时间: 2018-11-21 17:00

作者: 魔都黑马少年梦 时间: 2018-11-22 16:52

欢迎光临黑马程序员技术交流社区 (http://bbs.itheima.com/)

黑马程序员IT技术论坛 X3.2