Merge pull request #70 from annapovey/recognize_model_encoder_fix

fix model encoder to use a different model than the default
k2-fsa · Aug 29, 2024 · 1199e12 · 1199e12
2 parents 973009a + 73f1267
commit 1199e12
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(