Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "9c93f07c35ec896ea6d7df9ac8e604de7c263d84"
Commit 519ad098 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 264867908
parent 640ff472
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/transformer/v2/beam_search.py
View file @ 519ad098
...@@ -63,7 +63,7 @@ def sequence_beam_search(symbols_to_logits_fn, ...@@ -63,7 +63,7 @@ def sequence_beam_search(symbols_to_logits_fn,
alpha, alpha,
max_decode_length, max_decode_length,
eos_id, eos_id,
padded_decode, padded_decode=False,
dtype="float32"): dtype="float32"):
"""Search for sequence of subtoken ids with the largest probability. """Search for sequence of subtoken ids with the largest probability.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment