tensorflow持久化數據格式(2)

所有變量的值都存在下圖兩個變量中。
文件解釋
查看這兩個文件的內容:

# 查看變量信息

# 讀取checkpoint文件,.data,.index可以省略
reader = tf.train.NewCheckpointReader(r'E:\ZC\procedure\tensorflow_gpu\model\model.ckpt')
# 獲取變量名,變量的維度,變量的值
global_variable = reader.get_variable_to_shape_map()
for variable_name in global_variable:
    # 獲取變量的取值
    print(+variable_name, global_variable[variable_name], reader.get_tensor(variable_name))

結果
checkpoint文件是tf.train.Saver類自動生成自動維護的,名稱也是固定的。
CheckpointState類型的定義:
message ChceckpointState
{
string model_checkpoint_path=1;//最新的模型文件的文件名
repeated string all_model_checkpoint_paths=2;//所有模型文件的文件名

}
在這裏插入圖片描述

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