有時候會需要通過從儲存下來的ckpt檔案來觀察其儲存下來的訓練完成的變數值。

ckpt檔名列表：（一般是三個檔案）

xxxxx.ckpt.data-00000-of-00001

xxxxx.ckpt.index

xxxxx.ckpt.meta

import os
from tensorflow.python import pywrap_tensorflow

checkpoint_path = os.path.join("資料夾路徑","xxxxx.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) #tf.train.NewCheckpointReader
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
	print("tensor_name: ",key)
	# print(reader.get_tensor(key))

以上這篇tensorflow實現列印ckpt模型儲存下的變數名稱及變數值就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。