tensorflow實(shí)現(xiàn)打印ckpt模型保存下的變量名稱及變量值
有時(shí)候會(huì)需要通過從保存下來(lái)的ckpt文件來(lái)觀察其保存下來(lái)的訓(xùn)練完成的變量值。
ckpt文件名列表:(一般是三個(gè)文件)
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實(shí)現(xiàn)打印ckpt模型保存下的變量名稱及變量值就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python容器使用的5個(gè)技巧和2個(gè)誤區(qū)總結(jié)
在本篇文章里小編給大家整理的是關(guān)于Python容器使用的5個(gè)技巧和2個(gè)誤區(qū)的相關(guān)知識(shí)點(diǎn)內(nèi)容,需要的朋友們學(xué)習(xí)下。2019-09-09小白如何入門Python? 制作一個(gè)網(wǎng)站為例
以制作一個(gè)網(wǎng)站為例,聊一聊小白如何入門Python,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2018-03-03Python 循環(huán)讀取數(shù)據(jù)內(nèi)存不足的解決方案
這篇文章主要介紹了Python 循環(huán)讀取數(shù)據(jù)內(nèi)存不足的解決方案,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2021-05-05Django數(shù)據(jù)庫(kù)遷移常見使用方法
這篇文章主要介紹了Django數(shù)據(jù)庫(kù)遷移常見使用方法,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-11-11OpenCV特征提取與檢測(cè)之Shi-Tomasi角點(diǎn)檢測(cè)器
在角點(diǎn)檢測(cè)的世界里哈瑞斯無(wú)疑是最重要的方法之一,但Shi-Tomasi作為改進(jìn)的算法也有很大應(yīng)用場(chǎng)景,尤其是動(dòng)態(tài)跟蹤用的還比較多,這篇文章主要給大家介紹了關(guān)于OpenCV特征提取與檢測(cè)之Shi-Tomasi角點(diǎn)檢測(cè)器的相關(guān)資料,需要的朋友可以參考下2021-08-08python加密解密庫(kù)cryptography使用openSSL生成的密匙加密解密
這篇文章主要介紹了python加密解密庫(kù)cryptography使用openSSL生成的密匙加密解密,需要的朋友可以參考下2020-02-02