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

代碼比較簡單，就是取每一個輪次訓練後損失最小的那一次保存模型，並進行後續加載