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