3

tensorflow で提供される seq2seq.py ライブラリを使用して seq2seq モデルを構築します。何かをトレーニングする前に、トレーニングされていないモデルのグラフ ネットワークをテンソルボードで視覚化したかったのですが、これを表示したくありません。

私の問題を再現するための最小限の例を以下に示します。なぜこれがうまくいかないのか、誰にも分かりますか？トレーニング後のモデルのグラフのみを視覚化できますか?

import tensorflow as tf
import numpy as np
from tensorflow.models.rnn import rnn_cell
from tensorflow.models.rnn import seq2seq

encoder_inputs = []
decoder_inputs = []

for i in xrange(350):  
    encoder_inputs.append(tf.placeholder(tf.float32, shape=[None,2],
                                              name="encoder{0}".format(i)))

for i in xrange(45):
    decoder_inputs.append(tf.placeholder(tf.float32, shape=[None,22],
                                         name="decoder{0}".format(i)))

size = 512 # number of hidden units
num_layers = 2 # Number of LSTMs
single_cell = rnn_cell.BasicLSTMCell(size)
cell = rnn_cell.MultiRNNCell([single_cell] * num_layers)
model = seq2seq.basic_rnn_seq2seq(encoder_inputs, decoder_inputs,cell)

sess = tf.Session()
sess.run(tf.variables.initialize_all_variables())
summary_writer = tf.train.SummaryWriter('/path/to/log', graph_def = sess.graph_def)
4

1 に答える 1

3

これは、firefox ブラウザーでグラフの視覚化が機能しないというバグに関連しているようです。可能であれば、chrome または safari を使用してみてください。

https://github.com/tensorflow/tensorflow/issues/650

于 2016-01-20T20:33:05.567 に答える