模型保存,先要创建一个Saver对象:saver=tf.train.Saver(),
max_to_keep 是用来设置保存模型的个数,默认为5,即保存最近的五个模型,saver=tf.train.Saver(max_to_keep=0)
如果你只想保存最后一代的模型,则只需要将max_to_keep设置为1:saver=tf.train.Saver(max_to_keep=1)
创建完saver对象后,就可以保存训练好的模型了:saver.save(sess,'ckpt/mnist.ckpt',global_step=step)
模型的恢复用的是restore()函数,它需要两个参数restore(sess, save_path),save_path指的是保存的模型路径。
model_file=tf.train.latest_checkpoint('ckpt/') 自动获取最后一次保存的模型
saver.restore(sess,model_file)
原博客:https://www.cnblogs.com/denny402/p/6940134.html