tensorflow模型的保存与恢复

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

|--models
| |--checkpoint
| |--.meta
| |--.data
| |--.indexcode

.meta保存的是图的结构blog

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

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

 

tensorflow经常使用的模型保存方法: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

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

相关文章
相关标签/搜索