加速Keras LSTM

H.H*_*ter 3 python-3.x keras tensorflow

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

非常感谢你提前

H.B*_*ari 6

将标准替换LSTMCuDNNLSTM.你可以从中导入,

from keras.layers import  CuDNNLSTM 
Run Code Online (Sandbox Code Playgroud)