TensorFlow:保存和提取模型

描述

在訓練神經網絡模型的時候,當模型訓練完之後,確切地說當訓練的session關閉之後,我們訓練出來的模型參數會全部丟失,從而無法有效複用模型,而TensorFlow中提供了很好地保存模型和提取模型的方法。

 

結果

如果你搭建的網絡模型沒有問題的話,那麼在對應的文件目錄下將會看到16個文件,TensorFlow只會保存最近5次的模型,每一個模型會有三個文件,外加一個checkpoint文件,下圖我的一個示例:

 

各文件說明

checkpoint文件保存了一個目錄下所有的模型文件列表,這個文件是tf.train.Saver類自動生成且自動維護的。

.meta文件保存了TensorFlow計算圖的結構,可以理解爲神經網絡的網絡結構 ,TensorFlow通過元圖(MetaGraph)來記錄計算圖中節點的信息以及運行計算圖中節點所需要的元數據。

.data-00000-of-00001文件保存了TensorFlow程序中每一個變量的取值,這個文件是通過SSTable格式存儲的,可以大致理解爲就是一個(key,value)列表。

.index是對應模型的索引文件

 

參考:https://www.jianshu.com/p/c3a7f5c47b83

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章