廢話不多說,直接看代碼吧!
import osfrom tensorflow.python import pywrap_tensorflowmodel_dir="/xxxxxxxxx/model.ckpt" #checkpoint的文件位置# Read data from checkpoint filereader = pywrap_tensorflow.NewCheckpointReader(model_dir)var_to_shape_map = reader.get_variable_to_shape_map()# Print tensor name and valuesfor key in var_to_shape_map: print("tensor_name: ", key) #輸出變量名 print(reader.get_tensor(key)) #輸出變量值
輸出結果:
這里只輸出了變量名
以上這篇TensorFlow 輸出checkpoint 中的變量名與變量值方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答