tensorflow的模型保存于读取

源自  https://www.cnblogs.com/denny402/p/6940134.html

在模型的保存部分有两个重要的函数,

第一个  :  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()来自动获取最后一次保存的模型。

 

 

 

 

 

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

          ps  :  open (w--覆盖,a-- 不覆盖)

 

1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md或论文文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。 5、资源来自互联网采集,如有侵权,私聊博主删除。 6、可私信博主看论文后选择购买源代码。 1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md或论文文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。 5、资源来自互联网采集,如有侵权,私聊博主删除。 6、可私信博主看论文后选择购买源代码。 1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md或论文文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。 5、资源来自互联网采集,如有侵权,私聊博主删除。 6、可私信博主看论文后选择购买源代码。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值