Python 3.x 加速Keras LSTM
我正在尝试使用Keras训练LSTM,但训练速度非常慢(我有gpu Nvidia Titan X)。有人知道我如何加快训练速度吗Python 3.x 加速Keras LSTM,python-3.x,tensorflow,keras,Python 3.x,Tensorflow,Keras,我正在尝试使用Keras训练LSTM,但训练速度非常慢(我有gpu Nvidia Titan X)。有人知道我如何加快训练速度吗 非常感谢您提前将标准LSTM替换为CuDNNLSTM。你可以从 from keras.layers import CuDNNLSTM 很高兴知道。但请注意,它不支持掩蔽、自定义激活或退出。但是要快得多!从1080 ti上的43秒变为每个历元19秒!很好:)谢谢你!
非常感谢您提前将标准
LSTM
替换为CuDNNLSTM
。你可以从
from keras.layers import CuDNNLSTM
很高兴知道。但请注意,它不支持掩蔽、自定义激活或退出。但是要快得多!从1080 ti上的43秒变为每个历元19秒!很好:)谢谢你!