tensorFlow 讀取 .ckpt 斷點文件權值

from tensorflow.python import pywrap_tensorflow
model_reader = pywrap_tensorflow.NewCheckpointReader(r"model.ckpt-1000") # ckpt文件
var_dict = model_reader.get_variable_to_shape_map()
variable = model_reader.get_tensor('fc1/w')  # 作用域/變量名
for key in var_dict:
    print("fc1: ", key)
    print(model_reader.get_tensor(key))

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章