모두. 하나의 LSTM 모델 (https://github.com/roatienza/Deep-Learning-Experiments/tree/master/Experiments/Tensorflow/RNN)에서 고정 된 모델을 생성하고 싶습니다. 필자의 선택에 따라 마지막 예측 노드를 고정시키고 "bazel-bin/tensorflow/python/tools/freeze_graph --input_binary = true --input_graph = model_20170913/model.pb --input_checkpoint = model_20170913/model.ckpt - output_graph = model_20170913/frozen_graph.pb --output_node_names = ArgMax_52 "(ArgMax_52는 마지막 기본 노드 이름입니다). 그러나, 하나의 통고 "const op에 0 변수를 변환." (명령의 결과 고정). 자, node_name이 output_node_name으로되어야한다는 것에 대해서는 잘 모르겠다.LSTM 모델의 output_node_names 란 무엇입니까?
0
A
답변
0
위에서 언급 한 것처럼 "lstm_prediction"은 output_node_name입니다. 그리고 Tensorboard는 그래프를 이해하는데 많은 도움을줍니다.
0
"lstm_prediction"을 output_node_name으로 사용할 수있는 경우이를 사용하여 고정 된 .pb 파일을 내보낼 수 있습니다.
저장하기 전에 모든 노드 이름을 인쇄합니다. 놀랍게도이 노드에서 "ArgMax"와 "ArgMax_i"(i> = 1 && i <= 52)는 "bazel-bin/tensorflow/python/tools/freeze_graph --input_binary = true --input_graph = model_20170913 /model.pb --input_checkpoint = model_20170913/model.ckpt --output_graph = model_20170913/frozen_graph.pb --output_node_names = ArgMax "는 정지 모델을 생성 할 수 있습니다! –
코드 "return tf.add (tf.matmul (outputs [-1], weights [ 'out']), bias [ 'out'], name ="lstm_prediction ")"를 변경했습니다. "lstm_prediction"을 output_node_name으로 간주 할 수 있습니까? –