tensorflow模型的保存与恢复

模型保存后产生四个文件,分别是:

|--models
| |--checkpoint
| |--.meta
| |--.data
| |--.index

.meta保存的是图的结构

checkpoint文件是个文本文件,里面记录了保存的最新的checkpoint文件以及其它checkpoint文件列表。

.data和.index保存的是变量值。

 

tensorflow常用的模型保存方法:

best_str = ''
if best_loss is None or valid_loss < best_loss:
  best_loss = valid_loss
  best_str = ', best!'
  early_stop_i = 0
  saver.save(sess, '%s/model_best.tf' % FLAGS.logdir)
else:
  early_stop_i += 1

代码比较简单,就是取每个轮次训练后损失最小的那一次保存模型,并进行后续加载

 

posted on 2019-10-16 17:21  tuzhuo  阅读(295)  评论(0编辑  收藏  举报