【发布时间】：2018-07-04 02:07:06
【问题描述】：
我正在尝试使用 Keras 训练 LSTM，但训练速度非常慢（我有 gpu Nvidia Titan X）。有人知道我怎样才能加快训练速度吗？
非常感谢您
【问题讨论】：
标签： python-3.x tensorflow keras
【发布时间】：2018-07-04 02:07:06
【问题描述】：
将标准 LSTM 替换为 CuDNNLSTM。你可以导入它，
from keras.layers import CuDNNLSTM
【讨论】：
-
很高兴知道。请注意，它不支持屏蔽、自定义激活或退出。但是要快得多！
-
从 1080 ti 的 43 秒缩短到每个 epoch 的 19 秒！很好:) 谢谢！