Tensorflow: 從checkpoint文件中讀取tensor方式
更新時(shí)間：2020年02月10日 14:49:57 作者：noirblack
今天小編就為大家分享一篇Tensorflow: 從checkpoint文件中讀取tensor方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
在使用pre-train model時(shí)候，我們需要restore variables from checkpoint files.
經(jīng)常出現(xiàn)在checkpoint 中找不到”Tensor name not found”.
這時(shí)候需要查看一下ckpt中到底有哪些變量
import os
from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join(model_dir, "model.ckpt")
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
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))
可以顯示ckpt中的tensor名字和值，當(dāng)然也可以用pycharm調(diào)試。
以上這篇Tensorflow: 從checkpoint文件中讀取tensor方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Django框架的中的setting.py文件說(shuō)明詳解
這篇文章主要介紹了Django框架的中的setting.py文件說(shuō)明詳解，這個(gè)文件包含了所有有關(guān)這個(gè)Django項(xiàng)目的配置信息，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-10-10
Python中用altzone()方法處理時(shí)區(qū)的教程
這篇文章主要介紹了Python中用altzone()方法處理時(shí)區(qū)的教程,是Python入門(mén)中的基礎(chǔ)知識(shí),需要的朋友可以參考下2015-05-05
這篇文章主要介紹了Python pickle模塊用法實(shí)例,python的pickle模塊實(shí)現(xiàn)了基本的數(shù)據(jù)序列和反序列化,需要的朋友可以參考下2015-04-04
一篇文章教你用Python實(shí)現(xiàn)一個(gè)學(xué)生管理系統(tǒng)
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)學(xué)生管理系統(tǒng)，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-09-09
詳解python數(shù)據(jù)結(jié)構(gòu)之隊(duì)列Queue
這篇文章主要介紹了python數(shù)據(jù)結(jié)構(gòu)之隊(duì)列Queue,文中有非常詳細(xì)的代碼示例,對(duì)正在學(xué)習(xí)python的小伙伴們有很好的幫助,需要的朋友可以參考下2021-05-05
python如何實(shí)現(xiàn)常用的五種排序算法詳解
排序有很多種實(shí)現(xiàn)方法，比如冒泡排序、選擇排序、歸并排序、希爾排序、快速排序、插入排序、堆排序、基數(shù)排序等,這篇文章主要給大家介紹了關(guān)于python如何實(shí)現(xiàn)常用的五種排序算法,需要的朋友可以參考下2021-08-08
這篇文章主要介紹了如何通過(guò)python檢查文件是否被占用，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12
Python實(shí)現(xiàn)加載及解析properties配置文件的方法
這篇文章主要介紹了Python實(shí)現(xiàn)加載及解析properties配置文件的方法,結(jié)合實(shí)例形式分析了Python針對(duì)properties配置文件的加載、讀取及解析相關(guān)操作技巧,需要的朋友可以參考下2018-03-03
pycharm遠(yuǎn)程連接服務(wù)器調(diào)試tensorflow無(wú)法加載問(wèn)題
最近打算在win系統(tǒng)下使用pycharm開(kāi)發(fā)程序，并遠(yuǎn)程連接服務(wù)器調(diào)試程序，其中在import tensorflow時(shí)報(bào)錯(cuò)，本文就來(lái)介紹一下如何解決，感興趣的可以了解一下2021-06-06
Python開(kāi)發(fā)中的Nonetype類(lèi)型詳解
這篇文章主要介紹了Python開(kāi)發(fā)中的Nonetype類(lèi)型詳解,None有自己的數(shù)據(jù)類(lèi)型NoneType,你可以將None復(fù)制給任何變量,但是你不能創(chuàng)建其他NoneType對(duì)象,需要的朋友可以參考下2023-12-12