python-3.x – 加速Keras LSTM [复制]

发布时间：2020-12-20 11:06:00 所属栏目：Python 来源：网络整理

导读：参见英文答案 keras LSTM layer takes too long to train????????????????????????????????????1个我正在尝试使用Keras训练LSTM,但训练非常缓慢(我有gpu Nvidia Titan X).有人知道我怎样才能加快我的训练速度？非常感谢你提前解决方法将标准LSTM替换为Cu

参见英文答案 > keras LSTM layer takes too long to train????????????????????????????????????1个
我正在尝试使用Keras训练LSTM,但训练非常缓慢(我有gpu Nvidia Titan X).有人知道我怎样才能加快我的训练速度？

非常感谢你提前

解决方法

将标准LSTM替换为CuDNNLSTM.你可以从中导入,

from keras.layers import  CuDNNLSTM

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!