TensorFlow将meta转化为summary，从ckpt查看tensor的name

meta转化为summary用tensorboard查看

import tensorflow as tf
from tensorflow.python.platform import gfile

graph = tf.get_default_graph()
graphdef = graph.as_graph_def()
_ = tf.train.import_meta_graph('meta的路径/xxxx.meta')
summary = tf.summary.FileWriter('summary的路径', graph)

从ckpt查看tensor的name

from tensorflow.python import pywrap_tensorflow
 
checkpoint_path='ckpt的路径/xxxx.ckpt'
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)

来源：CSDN

作者：我怎么那么好看！

链接：https://blog.csdn.net/qq_34500588/article/details/103706216

标签