tensorflow模型的保存与恢复

Posted 2020-11-16 acm-icpcer

模型保存后产生四个文件，分别是：

|--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

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