小编Sha*_*awn的帖子

如何在tensorflow中找到给定.ckpt.meta文件的输出节点名称

到目前为止,我可以得到给定任何 .ckpt.meta 文件的所有节点名称的列表,但我想知道是否有系统的方法可以从列表中找出输出节点名称。

import tensorflow as tf

tf.reset_default_graph()
with tf.Session() as sess:
    saver = tf.train.import_meta_graph('mymodel.ckpt.meta')
    graph_def = tf.get_default_graph().as_graph_def()
    node_list=[n.name for n in graph_def.node]
Run Code Online (Sandbox Code Playgroud)

python machine-learning deep-learning tensorflow

5
推荐指数
1
解决办法
3463
查看次数