我正在尝试使用Keras训练LSTM,但训练非常缓慢(我有gpu NvIDia Titan X).有人知道我怎样才能加快我的训练速度?
非常感谢你提前
解决方法 将标准LSTM替换为CuDNNLSTM.你可以从中导入,from keras.layers import CuDNNLSTM总结
以上是内存溢出为你收集整理的python-3.x – 加速Keras LSTM [复制]全部内容,希望文章能够帮你解决python-3.x – 加速Keras LSTM [复制]所遇到的程序开发问题。
如果觉得内存溢出网站内容还不错,欢迎将内存溢出网站推荐给程序员好友。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)