tensorflow查看ckpt各節點名稱實例

 更新時間:2020-01-26 12:00:43   作者:佚名   我要評論(0)

運行下列腳本,可以打印出模型各個節點變量的名稱:


from tensorflow.python import pywrap_tensorflow
import os

checkpoint_path=os.path.join('model.ckpt-13

運行下列腳本,可以打印出模型各個節點變量的名稱:

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各節點名稱實例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

您可能感興趣的文章:

  • 從訓練好的tensorflow模型中打印訓練變量實例
  • TensorFlow 模型載入方法匯總(小結)
  • tensorflow實現打印ckpt模型保存下的變量名稱及變量值
  • TensorFlow——Checkpoint為模型添加檢查點的實例
  • tensorflow 實現打印pb模型的所有節點

相關文章

最新評論

老快3投注技巧 股票涨跌的本质 加拿大快乐8官方开奖结果 湖北快三官网开奖结果 吉林快3遗漏号 贵州茅台股票分析报告 广东南粤风采36选7预测 吉林十一选五的走势手机版 青海快3今天电子走势图 同花顺炒股软件下载 山西快乐10分走势图派彩电子 1000元怎么玩股 河北十一选五走势图定牛 双色球下期预测专家最准确 山东11选5推荐任三预测 南京股票期货配资 11选5最常出的三号