mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-05 07:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix decoding the gigaspeech dataset.

Browse Source 
We have to use the decoder/joiner networks for the GigaSpeech dataset.
This commit is contained in:
Fangjun Kuang 2022-04-18 15:23:07 +08:00
parent e32641d1df
commit a31207f5b3
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/ASR/pruned_transducer_stateless3/decode-giga.py
View File

@ -520,6 +520,11 @@ def main():
model.eval()
model.device = device
# In beam_search.py, we are using model.decoder() and model.joiner(),
# so we have to switch to the branch for the GigaSpeech dataset.
model.decoder = model.decoder_giga
model.joiner = model.joiner_giga
if params.decoding_method == "fast_beam_search":
decoding_graph = k2.trivial_graph(params.vocab_size - 1, device=device)
else: