fix model encoder to use a different model than the default

k2-fsa · Jul 28, 2024 · 73f1267 · 73f1267
1 parent 7c452ed
commit 73f1267
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/examples/libriheavy/tools/recognize.py b/examples/libriheavy/tools/recognize.py
@@ -181,10 +181,13 @@ def decode_one_batch(
     supervisions = batch["supervisions"]
     feature_lens = supervisions["num_frames"].to(device)
 
-    encoder_out, encoder_out_lens = model.encoder(
-        features=feature,
-        feature_lengths=feature_lens,
-    )
+    #encoder_out, encoder_out_lens = model.encoder(
+    #    features=feature,
+    #    feature_lengths=feature_lens,
+    #)
+
+    # to use a different model than the default this fix is needed 
+    encoder_out, encoder_out_lens = model.encoder(feature,feature_lens,)
 
     if params.decoding_method == "greedy_search":
         res = greedy_search_batch(