mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-02 21:54:18 +00:00
Minor fixes
This commit is contained in:
parent
18dd5678d9
commit
dbfe8fbb1a
@ -90,7 +90,7 @@ def fast_beam_search(
|
||||
logits = logits.squeeze(1).squeeze(1)
|
||||
log_probs = logits.log_softmax(dim=-1)
|
||||
decoding_streams.advance(log_probs)
|
||||
decoding_streams.detach()
|
||||
decoding_streams.terminate_and_flush_to_atreams()
|
||||
lattice = decoding_streams.format_output(encoder_out_lens.tolist())
|
||||
|
||||
best_path = one_best_decoding(lattice)
|
||||
|
Loading…
x
Reference in New Issue
Block a user