從訓練好的tensorflow模型中打印訓練變量實例
更新時間:2020年01月20日 09:38:16 作者:aiseu001
今天小編就為大家分享一篇從訓練好的tensorflow模型中打印訓練變量實例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
從tensorflow 訓練后保存的模型中打印訓變量:使用tf.train.NewCheckpointReader()
import tensorflow as tf reader = tf.train.NewCheckpointReader('path/alexnet/model-330000') dic = reader.get_variable_to_shape_map() print dic
打印變量
w = reader.get_tensor("fc1/W") print type(w) print w.shape print w[0]
以上這篇從訓練好的tensorflow模型中打印訓練變量實例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
python教程網(wǎng)絡爬蟲及數(shù)據(jù)可視化原理解析
這篇文章主要為大家介紹了python教程中網(wǎng)絡爬蟲及數(shù)據(jù)可視化原理的示例解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步早日升職加薪2021-10-10