我正在通过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)
答案 0 :(得分:0)
bidirectional_dynamic_rnn比static_bidirectional_rnn快。