断电续训
1. 关键处理:加入ckpt操作:
ckpt = tf.train.get_checkpoint_state(MODEL_SAVE_PATH)
if ckpt and ckpt.model_checkpoint_path:
saver.restore(sess,ckpt.model_checkpoint_path)
注解:
1. tf.train.get_checkpoint_state(checkpoint_dir,laster_filename=None) 该函数表明如果断点文件夹中包含有效断点状态文件,则返回该文件。
参数说明:checkpoint_dir:表示存储断点文件的目录
latest_filename=None:断点文件的可选名称,默认为"checkpoint"
2. saver.restore(sess,ckpt.model_checkpoint_path)
该函数表示恢复当前会话,将ckpt中的值赋给w和b。
参数说明:sess: 表示当前会话,之前保存的结果将被加载入这个会话
ckpt.model_checkpoint_path:表示模型存储的位置,不需要提供模型 的名字,它会去查看checkpoint文件,看看最新的是谁,叫做什么。
注:本文章通过观看北京大学曹健老师的Tensorflow视频,笔记总结而来的。