Tensorflow: 從checkpoint檔案中讀取tensor方式
阿新 • • 發佈:2020-02-11
在使用pre-train model時候,我們需要restore variables from checkpoint files.
經常出現在checkpoint 中找不到”Tensor name not found”.
這時候需要檢視一下ckpt中到底有哪些變數
import os from tensorflow.python import pywrap_tensorflow checkpoint_path = os.path.join(model_dir,"model.ckpt") # Read data from checkpoint file reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map = reader.get_variable_to_shape_map() # Print tensor name and values for key in var_to_shape_map: print("tensor_name: ",key) print(reader.get_tensor(key))
可以顯示ckpt中的tensor名字和值,當然也可以用pycharm除錯。
以上這篇Tensorflow: 從checkpoint檔案中讀取tensor方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。