所有變量的值都存在下圖兩個變量中。
查看這兩個文件的內容:
# 查看變量信息
# 讀取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;//所有模型文件的文件名
}