TensorFlow 輸出checkpoint 中的變量名與變量值方式
更新時間：2020年02月11日 14:40:54 作者：qq_32799915
今天小編就為大家分享一篇TensorFlow 輸出checkpoint 中的變量名與變量值方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
廢話不多說，直接看代碼吧！
import os
from tensorflow.python import pywrap_tensorflow
model_dir="/xxxxxxxxx/model.ckpt" #checkpoint的文件位置
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(model_dir)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
print("tensor_name: ", key) #輸出變量名
print(reader.get_tensor(key)) #輸出變量值
輸出結(jié)果：
這里只輸出了變量名
以上這篇TensorFlow 輸出checkpoint 中的變量名與變量值方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
pandas數(shù)據(jù)框,統(tǒng)計某列數(shù)據(jù)對應(yīng)的個數(shù)方法
下面小編就為大家分享一篇pandas數(shù)據(jù)框,統(tǒng)計某列數(shù)據(jù)對應(yīng)的個數(shù)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
python神經(jīng)網(wǎng)絡(luò)MobileNetV2模型的復(fù)現(xiàn)詳解
這篇文章主要為大家介紹了python神經(jīng)網(wǎng)絡(luò)MobileNetV2模型的復(fù)現(xiàn)詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-05-05
這篇文章主要為大家詳細介紹了python實現(xiàn)圖像邊緣檢測，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-08-08
TensorFlow中tf.batch_matmul()的用法
這篇文章主要介紹了TensorFlow中tf.batch_matmul()的用法，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-06-06
今天小編就為大家分享一篇python 使用plt畫圖,去除圖片四周的白邊方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
這篇文章主要給大家介紹了關(guān)于python自動生成證件號的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-01-01
在本篇內(nèi)容里小編給大家整理的是關(guān)于python中取絕對值簡單方法，需要的朋友們可以學(xué)習(xí)下。2020-07-07
這篇文章主要為大家詳細介紹了python tornado微信開發(fā)入門代碼，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-08-08
這篇文章主要為大家詳細介紹了python編程實現(xiàn)維吉尼亞算法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-03-03
這篇文章主要介紹了Django 限制訪問頻率的思路詳解，本文給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2019-12-12