From a31207f5b3670f5eb0287d24e87bce9e324ed254 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Mon, 18 Apr 2022 15:23:07 +0800
Subject: [PATCH] Fix decoding the gigaspeech dataset.

We have to use the decoder/joiner networks for the GigaSpeech dataset.
---
 .../ASR/pruned_transducer_stateless3/decode-giga.py          | 5 +++++
 1 file changed, 5 insertions(+)

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: