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 has been 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.
The text was updated successfully, but these errors were encountered:
Keras 2.1.0 offers
keras.layers.CuDNNGRU
, a drop-in replacement forkeras.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.The text was updated successfully, but these errors were encountered: