tensorflow保存文件格式總結

一、背景

tensorflow還是非常靈活的,但是有門檻的,門檻有兩點api比較多,理解tensorflow的思維(先建立圖,然後建立session,然後運算,pytorch是不用這樣),這篇文章說說保存tensorflow保存的模型

Tensorflow的保存分爲四種

  • 1. checkpoint模式;
  • 2. saved_model模式(包含pb文件和variables);
  • 3. 純pb模式;(只有一個pb文件)
  • 4.keras的 h5 模式

1. checkpoint模式

# 一個完整的模型保存包含下面四個文件
 checkpoint: 具有最近檢查點列表的協議緩存區
.data: 保存模型中的變量
.index: 標識檢查點
.meta: 保存模型中計算圖的結構信息

 

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