You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Sep 25, 2024. It is now read-only.
Keras 2.1.0 offers keras.layers.CuDNNGRU, a drop-in replacement for keras.layers.GRU, that's backed by an optimized implementation from CuDNN.
Speedup for the RNN computation observed in nvprof is 6x, similar to what's reported elsewhere about CuDNNGRU. Note that it needs TensorFlow, CuDNN and an NVIDIA GPU. For CPU usage it has to be replaced by ordinary GRU layer. Saved weights are compatible. In practice it works well.