Commit 7161c820 authored by Beomsu Kim's avatar Beomsu Kim Committed by Rasmus Munk Larsen
Browse files

Fix contrib.seq2seq.BeamSearchDecoder #9855 (#9875) 

* Shielded top_k input with min

Fixes #9855.

* Fix docstring note about BeamSearch _get_scores

* Change next_beam_size to tensor

* Calculate number of available beams before cond
parent fca253c2
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment