断点续训

断电续训

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视频，笔记总结而来的。