Seq2Seq预测速度很慢

时间:2017-12-30 03:48:13

标签: tensorflow deep-learning

我正在尝试使用LSTM在tensorflow中实现Seq2Seq模型(从头开始,没有rnn单元格),模型工作正常,但一句话的预测时间对我来说很慢。约2 - > 6秒一句话。这是正常的吗?

我的模特:

2 LSTM for encode

2 LSTM用于解码

注意机制

词汇量:400k

单词矢量维度:300

预测代码完全在CPU上。

我读了一些论文,但他们没有提供预测速度。谢谢!

0 个答案:

没有答案