一、背景
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: 保存模型中計算圖的結構信息