tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值
更新時間：2020年01月04日 11:00:06 作者：Shwan_Ma
今天小編就為大家分享一篇tensorflow實現(xiàn)打印ckpt模型保存下的變量名稱及變量值，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
有時候會需要通過從保存下來的ckpt文件來觀察其保存下來的訓(xùn)練完成的變量值。
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模型保存下的變量名稱及變量值就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
在本文中我們給大家詳細(xì)講解了關(guān)于Python sys.argv使用方法以及注意事項，有此需要的讀者們跟著學(xué)習(xí)下。2019-05-05
今天小編就為大家分享一篇tensorflow2.0保存和恢復(fù)模型3種方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
Python容器使用的5個技巧和2個誤區(qū)總結(jié)
在本篇文章里小編給大家整理的是關(guān)于Python容器使用的5個技巧和2個誤區(qū)的相關(guān)知識點內(nèi)容，需要的朋友們學(xué)習(xí)下。2019-09-09
以制作一個網(wǎng)站為例，聊一聊小白如何入門Python，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-03-03
這篇文章主要介紹了如何使用Python監(jiān)控屏幕變化并截圖,文中通過代碼示例給大家講解的非常詳細(xì),并對大家的學(xué)習(xí)工作有一定的幫助,需要的朋友可以參考下2024-04-04
Python 循環(huán)讀取數(shù)據(jù)內(nèi)存不足的解決方案
這篇文章主要介紹了Python 循環(huán)讀取數(shù)據(jù)內(nèi)存不足的解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-05-05
這篇文章主要介紹了Django數(shù)據(jù)庫遷移常見使用方法,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-11-11
在角點檢測的世界里哈瑞斯無疑是最重要的方法之一，但Shi-Tomasi作為改進的算法也有很大應(yīng)用場景,尤其是動態(tài)跟蹤用的還比較多,這篇文章主要給大家介紹了關(guān)于OpenCV特征提取與檢測之Shi-Tomasi角點檢測器的相關(guān)資料,需要的朋友可以參考下2021-08-08
這篇文章主要介紹了詳解python算法之冒泡排序，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03
python加密解密庫cryptography使用openSSL生成的密匙加密解密
這篇文章主要介紹了python加密解密庫cryptography使用openSSL生成的密匙加密解密,需要的朋友可以參考下2020-02-02