small fix for decode.py

k2-fsa · Jan 3, 2024 · a232ceb · a232ceb
1 parent 8023493
commit a232ceb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/egs/librispeech/SSL/hubert/decode.py b/egs/librispeech/SSL/hubert/decode.py
@@ -482,7 +482,7 @@ def decode_one_batch(
             max_contexts=params.max_contexts,
             max_states=params.max_states,
             num_paths=params.num_paths,
-            ref_texts=sp.encode(supervisions["text"]),
+            ref_texts=sp.encode(batch["supervisions"]["text"]),
             nbest_scale=params.nbest_scale,
         )
         for hyp in sp.decode(hyp_tokens):