首页 > 解决方案 > 加速 Keras LSTM

问题描述

我正在尝试使用 Keras 训练 LSTM,但训练速度非常慢(我有 gpu Nvidia Titan X)。有人知道我怎样才能加快训练速度吗?

非常感谢您提前

标签: python-3.xtensorflowkeras

解决方案


将标准替换LSTMCuDNNLSTM. 您可以从以下位置导入它,

from keras.layers import  CuDNNLSTM 

推荐阅读