diff --git a/egs/librispeech/SSL/hubert/decode.py b/egs/librispeech/SSL/hubert/decode.py index 7df3c29634..1e96224ebb 100644 --- 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):