diff options
| author |  Katherine Wu <kathywu@google.com> | 2018-01-12 17:37:43 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-12 17:41:29 -0800 |
| commit | 4476407836c5f0de0b746269af1714a469a09c32 (patch) | |
| tree | 095bc609b310ec765377bbef46cc935f74964980 /tensorflow/contrib/cudnn_rnn | |
| parent | 7d1d45954834f39ca2986599c7f93ae3c552dedd (diff) | |
Fix dropout description of CudnnLSTM.
PiperOrigin-RevId: 181814918
Diffstat (limited to 'tensorflow/contrib/cudnn_rnn')
| -rw-r--r-- | tensorflow/contrib/cudnn_rnn/python/layers/cudnn_rnn.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/cudnn_rnn/python/layers/cudnn_rnn.py b/tensorflow/contrib/cudnn_rnn/python/layers/cudnn_rnn.py index 5d640d09b7..36fba917a8 100644 --- a/tensorflow/contrib/cudnn_rnn/python/layers/cudnn_rnn.py +++ b/tensorflow/contrib/cudnn_rnn/python/layers/cudnn_rnn.py @@ -176,8 +176,9 @@ class _CudnnRNN(base_layer.Layer): otherwise, it implies 'linear_input'. direction: the direction model that the model operates. Can be either 'unidirectional' or 'bidirectional' - dropout: dropout rate, a number between [0, 1]. Dropout is applied on - inputs of each layer. When set to 0, dropout is disabled. + dropout: dropout rate, a number between [0, 1]. Dropout is applied between + each layer (no dropout is applied for a model with a single layer). + When set to 0, dropout is disabled. seed: the op seed used for initializing dropout. See @{tf.set_random_seed} for behavior. dtype: tf.float16, tf.float32 or tf.float64 |