极慢的建筑物双向TF Lstm单元

时间:2018-10-22 08:22:23

标签: python tensorflow rnn

我正在通过Tensorflow构建双向lstm层,但是构建过程非常缓慢,对于相当小的网络,完成构建需要花费数小时的时间。 我的测试隐藏层大小为16;输入的嵌入向量大小为32;输入序列长度为20,词汇量为20000

tf的表现是否达到预期?

随附了我用来构建网络的部分,以获取更多信息:

cell_fw = tf.contrib.rnn.LSTMCell(hidden_layer_size)
cell_bw = tf.contrib.rnn.LSTMCell(hidden_layer_size)
(emb_encoder_inputs, fw_state, _) = tf.contrib.rnn.static_bidirectional_rnn(
          cell_fw, cell_bw, emb_encoder_inputs, dtype=tf.float32,
          sequence_length=article_lens)

1 个答案:

答案 0 :(得分:0)

bidirectional_dynamic_rnn比static_bidirectional_rnn快。