tensorflow模型(ckpt文件)的保存

知识点:tensorflow的模型保存和加载又叫tensorflow持久化
一.模型保存
1.首先来看一个重点函数

saver= tf.train.Saver(max_to_keep=None, keep_checkpoint_every_n_hours=None)

此为tensorflow保存模型函数,通常内部参数可为空。此函数默认保存最近5个模型,如果想要增加保存数量,可将max_to_keep函数赋数量值;要是想隔一段固定的时间保存模型可给keep_checkpoint_every_n_hours赋值;函数默认保存所有变量,如果你只想保存指定变量,可以通过指定variables/collections,在创建tf.train.Saver实例时,通过将需要保存的变量构造list或者dictionary,传入到Saver中,如saver = tf.train.Saver([w1,w2])表示只保存w1,w2。

2.下面直接上保存代码:

# tensorflow保存训练模型
import tensorflow as tf

w1 = tf.Variable(tf.random_normal(shape=[2]), name='w1')
w2 = tf.Variable(tf.random_normal(shape=[5]), name='w2')
# tensorflow默认只会保存最近的5个模型文件,如果你希望保存更多,可以通过max_to_keep来指定
saver = tf.train.Saver(max_to_keep=5, keep_checkpoint_every_n_hours=2)  # 保存最近5个模型,
  • 6
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值