tensorflow檢視ckpt各節點名稱例項
阿新 • • 發佈:2020-01-22
執行下列指令碼,可以打印出模型各個節點變數的名稱:
from tensorflow.python import pywrap_tensorflow import os checkpoint_path=os.path.join('model.ckpt-131805') reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map=reader.get_variable_to_shape_map() for key in var_to_shape_map: print 'tensor_name: ',key
checkpoint_path為自己的模型路徑
以上這篇tensorflow檢視ckpt各節點名稱例項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。