溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

tensorflow如何查看ckpt各節(jié)點名稱實例

發(fā)布時間：2021-05-08 10:10:27 來源：億速云 閱讀：336 作者：小新 欄目：開發(fā)技術(shù)

這篇文章主要介紹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é)點名稱實例”這篇文章的所有內(nèi)容，感謝各位的閱讀！希望分享的內(nèi)容對大家有幫助，更多相關(guān)知識，歡迎關(guān)注億速云行業(yè)資訊頻道！

向AI問一下細節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手