Как узнать правильную операцию ввода и вывода замороженного тензорного графа? - PullRequest
3 голосов
/ 25 апреля 2019

Я получил нейронную сеть, обученную Python, формат замороженного графа (.pb). Чтобы сделать вывод с использованием C-API (что требуется для этого особого случая), мне нужно выяснить имя операции ввода и вывода графика. Очевидно, что имена отличаются от тех, кто во время обучения. Граф имеет 1780 операций, следующие содержат в своем имени «вход» или «выход»:

ввод:

bn1a_branch2a/keras_learning_phase/input

input_1_7



output:

output_1/kernel

output_1/bias

output_1_1/kernel

output_1_1/bias

output_1_2/kernel

output_1_2/bias

output_1_3/kernel

output_1_3/bias

output_1_4/kernel

output_1_4/bias

output_1_5/kernel

output_1_5/bias

output_1_6/kernel

output_1_6/bias

output_1_7/kernel

output_1_7/kernel/read

output_1_7/bias

output_1_7/bias/read

output_1_7/MatMul

output_1_7/BiasAdd

Есть ли способ узнать фактическую операцию ввода и вывода замороженного графика?

1 Ответ

0 голосов
/ 29 апреля 2019

Хорошо для тех, кто ищет ответ: кажется, что единственным надежным способом является визуализация графика с использованием Tensorboard. Таким образом вы можете избежать любых ловушек, независимо от того, сколько операций включает ваш график.

...