最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

tensorflow實(shí)現(xiàn)讀取模型中保存的值 tf.train.NewCheckpointReader

 更新時(shí)間:2020年02月10日 11:10:06   作者:onexming  
今天小編就為大家分享一篇tensorflow實(shí)現(xiàn)讀取模型中保存的值 tf.train.NewCheckpointReader,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

使用tf.trian.NewCheckpointReader(model_dir)

一個(gè)標(biāo)準(zhǔn)的模型文件有一下文件, model_dir就是MyModel(沒有后綴)

checkpoint
Model.meta
Model.data-00000-of-00001
Model.index
import tensorflow as tf
import pprint # 使用pprint 提高打印的可讀性
NewCheck =tf.train.NewCheckpointReader("model")

打印模型中的所有變量

print("debug_string:\n")
pprint.pprint(NewCheck.debug_string().decode("utf-8"))

其中有3個(gè)字段, 分別是名字, 數(shù)據(jù)類型, shape

獲取變量中的值

print("get_tensor:\n")
pprint.pprint(NewCheck.get_tensor("D/conv2d/bias"))

print("get_variable_to_dtype_map\n")
pprint.pprint(NewCheck.get_variable_to_dtype_map())
print("get_variable_to_shape_map\n")
pprint.pprint(NewCheck.get_variable_to_shape_map())

以上這篇tensorflow實(shí)現(xiàn)讀取模型中保存的值 tf.train.NewCheckpointReader就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。

相關(guān)文章

最新評(píng)論

丰原市| 恩平市| 大庆市| 衡阳市| 巧家县| 武威市| 图木舒克市| 格尔木市| 中宁县| 贵南县| 嘉义县| 枞阳县| 安福县| 海宁市| 西乌珠穆沁旗| 商河县| 红原县| 钦州市| 连云港市| 丹寨县| 阿勒泰市| 佳木斯市| 海丰县| 霍州市| 建德市| 贵溪市| 泽库县| 邯郸市| 平山县| 新野县| 和田县| 新和县| 伽师县| 大姚县| 广南县| 宝应县| 肥东县| 泰州市| 宁德市| 习水县| 达拉特旗|