Internal change

PiperOrigin-RevId: 264867908

Internal change
PiperOrigin-RevId: 264867908
519ad098 · A. Unique TensorFlower · 640ff472 · 519ad098
Commit 519ad098 authored Aug 22, 2019 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/transformer/v2/beam_search.py official/transformer/v2/beam_search.py +1 -1

No files found.
--- a/official/transformer/v2/beam_search.py
+++ b/official/transformer/v2/beam_search.py
@@ -63,7 +63,7 @@ def sequence_beam_search(symbols_to_logits_fn,
                         alpha,
                         max_decode_length,
                         eos_id,
-                         padded_decode,
+                         padded_decode=False,
                         dtype="float32"):
  """Search for sequence of subtoken ids with the largest probability.