Как найти имя выходного узла данного файла .ckpt.meta в тензорном потоке

Question

Как найти имя выходного узла данного файла .ckpt.meta в тензорном потоке

1

Пока я могу получить список всех имен узлов, заданных любым файлом.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]

Shawn 27 июль 2018, в 01:21

Источник

1

Обычно полезно использовать пространства имен, чтобы упростить поиск. Также вы можете использовать tf.identity чтобы дать выходу операции имя по своему вкусу. Вы можете получить имя переменной с помощью graph_def.get_tensor_by_name('mynamespace/myvar:0') . Я еще не нашел лучшего пути.
DocDriven 27 июль 2018, в 09:04
0

Вы нашли решение?
user260826 19 сен. 2018, в 23:24

Теги:

python

tensorflow

machine-learning

deep-learning

1 ответ

Ещё вопросы

Обычно полезно использовать пространства имен, чтобы упростить поиск. Также вы можете использовать tf.identity чтобы дать выходу операции имя по своему вкусу. Вы можете получить имя переменной с помощью graph_def.get_tensor_by_name('mynamespace/myvar:0') . Я еще не нашел лучшего пути.

spicyramen · Answer 1 · 2018-09-19T22-16-00.000Z

Ты можешь попробовать:

[n.name for n in tf.get_default_graph().as_graph_def().node]