search

farizrahman4u / seq2seq

Sequence to Sequence Learning with Keras
GNU General Public License v2.0
3.17k stars 845 forks source link

Cudnn based LSTM Layers? #266

Closed 0xDaksh closed 4 years ago

0xDaksh commented 5 years ago

Hi, as you may have seen that keras has CuDNNLSTM and CuDNNGRU layers optimized for GPUs.

Is there any possibility that we can add that to this?

I tried but There's no CuDNNLSTMCell available, so I wasn't able to get it running. But any help would be appreciated.