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)) #輸出變量值
輸出結果:

這里只輸出了變量名
以上這篇TensorFlow 輸出checkpoint 中的變量名與變量值方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
使用Python的pencolor函數(shù)實現(xiàn)漸變色功能
這篇文章主要介紹了使用Python的pencolor函數(shù)實現(xiàn)漸變色功能,本文通過實例代碼給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2021-03-03
Python中函數(shù)帶括號和不帶括號的區(qū)別及說明
這篇文章主要介紹了Python中函數(shù)帶括號和不帶括號的區(qū)別及說明,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-11-11
關于Python調用百度語音合成SDK實現(xiàn)文字轉音頻的方法
這篇文章主要介紹了關于Python調用百度語音合成SDK實現(xiàn)文字轉音頻的方法,AipSpeech是語音合成的Python?SDK客戶端,為使用語音合成的開發(fā)人員提供了一系列的交互方法,需要的朋友可以參考下2023-07-07
pytorch神經(jīng)網(wǎng)絡之卷積層與全連接層參數(shù)的設置方法
今天小編就為大家分享一篇pytorch神經(jīng)網(wǎng)絡之卷積層與全連接層參數(shù)的設置方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
在Python的Django框架的視圖中使用Session的方法
這篇文章主要介紹了在Python的Django框架的視圖中使用Session的方法,包括相關的設置測試Cookies的方法,需要的朋友可以參考下2015-07-07

