tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值
更新時間:2020年01月04日 11:00:06 作者:Shwan_Ma
今天小編就為大家分享一篇tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
有時候會需要通過從保存下來的ckpt文件來觀察其保存下來的訓練完成的變量值。
ckpt文件名列表:(一般是三個文件)
xxxxx.ckpt.data-00000-of-00001
xxxxx.ckpt.index
xxxxx.ckpt.meta
import os
from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join("文件夾路徑", "xxxxx.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) #tf.train.NewCheckpointReader
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print("tensor_name: ", key)
# print(reader.get_tensor(key))
以上這篇tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
Python 循環(huán)讀取數(shù)據(jù)內存不足的解決方案
這篇文章主要介紹了Python 循環(huán)讀取數(shù)據(jù)內存不足的解決方案,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2021-05-05
python加密解密庫cryptography使用openSSL生成的密匙加密解密
這篇文章主要介紹了python加密解密庫cryptography使用openSSL生成的密匙加密解密,需要的朋友可以參考下2020-02-02

