Tensorflow-查看保存的变量及名字

最新推荐文章于 2021-01-30 10:45:59 发布

风吴痕

最新推荐文章于 2021-01-30 10:45:59 发布

阅读量8k

点赞数 4

分类专栏： tensorflow python 文章标签： tensorflow

本文链接：https://blog.csdn.net/wc781708249/article/details/78040735

版权

python 同时被 2 个专栏收录

95 篇文章 0 订阅

订阅专栏

tensorflow

91 篇文章 1 订阅

订阅专栏

参考：http://blog.csdn.net/u010698086/article/details/77916532

import tensorflow as tf
import os

logdir='./output/'

from tensorflow.python import pywrap_tensorflow
# checkpoint_path = os.path.join(model_dir, "model.ckpt-9999")
ckpt = tf.train.get_checkpoint_state(logdir)
reader = pywrap_tensorflow.NewCheckpointReader(ckpt.model_checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print("tensor_name: ", key)
    print(reader.get_tensor(key))