加速Keras LSTM

Question

3

我正在尝试使用Keras训练LSTM模型，但是训练速度非常慢（我的GPU是Nvidia Titan X）。有人知道如何加快训练速度吗？

非常感谢您的帮助。

- H.Hunter

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- H.Bukhari · Accepted Answer

将标准的LSTM替换为CuDNNLSTM。您可以从以下位置导入：

from keras.layers import  CuDNNLSTM