tensorflow模型保存与载入

用Tensorflow做一些事情,最近遇到的问题是训练到的模型怎么用?

网上搜了一下,最常用的就是使用tf.train.Saver()定义saver变量,然后分别使用saver.save()和saver.restore()载入。

保存代码:

saver = tf.train.Saver() 
with tf.Session() as sess: 
  sess.run(init) 
  save_path = saver.save(sess,"results/model") 

保存后,在results目录会生成四个文件:

  • checkpoint:该文件记录了模型文件的路径信息
  • model.data-00000-of-0001:网络权重信息
  • model.index:二进制文件,保存了模型中的变量参数信息
  • model.meta:保存了模型的计算图结构信息

tensorflow版本是1.14.0,若tf2.0,因为不再用计算图了,应该会有所不同,以后用到再说。

当然,还可以将中间的训练数据保存下来,而不仅保存最终状态:

saver.save(agent.sess, 'result\\model',global_step=i,write_meta_graph=True)

加上global_step后在迭代过程中保存模型,会在模型文件后加上“-i”,生成一系列的文件。载入的时候可以载

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值