diff options
Diffstat (limited to 'tensorflow/contrib/seq2seq/python/ops/loss.py')
-rw-r--r-- | tensorflow/contrib/seq2seq/python/ops/loss.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/seq2seq/python/ops/loss.py b/tensorflow/contrib/seq2seq/python/ops/loss.py index e14f07bc09..61852eda4f 100644 --- a/tensorflow/contrib/seq2seq/python/ops/loss.py +++ b/tensorflow/contrib/seq2seq/python/ops/loss.py @@ -44,8 +44,7 @@ def sequence_loss(logits, targets, weights, sequence. When using weights as masking set all valid timesteps to 1 and all padded timesteps to 0. average_across_timesteps: If set, sum the cost across the sequence - dimension and divide by the cost by the total label weight across - timesteps. + dimension and divide the cost by the total label weight across timesteps. average_across_batch: If set, sum the cost across the batch dimension and divide the returned cost by the batch size. softmax_loss_function: Function (labels-batch, inputs-batch) -> loss-batch |