tensorflow查看ckpt各節(jié)點名稱實例
更新時間:2020年01月21日 10:28:57 作者:深夜蟲鳴
今天小編就為大家分享一篇tensorflow查看ckpt各節(jié)點名稱實例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
運行下列腳本,可以打印出模型各個節(jié)點變量的名稱:
from tensorflow.python import pywrap_tensorflow import os checkpoint_path=os.path.join('model.ckpt-131805') reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path) var_to_shape_map=reader.get_variable_to_shape_map() for key in var_to_shape_map: print 'tensor_name: ',key
checkpoint_path為自己的模型路徑
以上這篇tensorflow查看ckpt各節(jié)點名稱實例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
處理python中多線程與多進程中的數(shù)據(jù)共享問題
這篇文章主要介紹了python中多線程與多進程中的數(shù)據(jù)共享問題,本文給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2019-07-07django實現(xiàn)更改數(shù)據(jù)庫某個字段以及字段段內數(shù)據(jù)
這篇文章主要介紹了django實現(xiàn)更改數(shù)據(jù)庫某個字段以及字段段內數(shù)據(jù),具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03