TensorFlow 輸出checkpoint 中的變數名與變數值方式
阿新 • • 發佈:2020-02-12
廢話不多說,直接看程式碼吧!
import os from tensorflow.python import pywrap_tensorflow model_dir="/xxxxxxxxx/model.ckpt" #checkpoint的檔案位置 # Read data from checkpoint file reader = pywrap_tensorflow.NewCheckpointReader(model_dir) 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)) #輸出變數值
輸出結果:
這裡只輸出了變數名
以上這篇TensorFlow 輸出checkpoint 中的變數名與變數值方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。