2015-11-18 1 views
3

Я построю модель seq2seq, используя библиотеку 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) 

ответ

3

Похоже, что это может быть связано с ошибкой, когда визуализация графа не работает в браузере firefox. Попробуйте использовать хром или сафари, если это возможно.

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

Смежные вопросы