我使用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)
看起来这可能与一个错误有关,在该错误中图形可视化在Firefox浏览器中不起作用。如果可能,请尝试使用chrome或safari。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句