diff --git a/egs/librispeech/ASR/pruned_transducer_stateless3/decode-giga.py b/egs/librispeech/ASR/pruned_transducer_stateless3/decode-giga.py index c04029c7f..e6a9a0aee 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless3/decode-giga.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless3/decode-giga.py @@ -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: