From cb15b9029cd1a89cf4a40718259b5d47f7256939 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 30 Sep 2023 21:29:03 +0800 Subject: [PATCH] Use kaldi-decoder --- .../ASR/conformer_ctc/jit_pretrained_decode_with_H.py | 2 +- .../ASR/conformer_ctc/jit_pretrained_decode_with_HL.py | 2 +- .../ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py | 2 +- requirements.txt | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py index 14c20c5b2..5783be0d1 100755 --- a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py +++ b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_H.py @@ -41,7 +41,7 @@ import kaldifeat import kaldifst import torch import torchaudio -from kaldi_hmm_gmm import DecodableCtc, FasterDecoder, FasterDecoderOptions +from kaldi_decoder import DecodableCtc, FasterDecoder, FasterDecoderOptions from torch.nn.utils.rnn import pad_sequence diff --git a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py index f02655468..da99a3465 100755 --- a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py +++ b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HL.py @@ -41,7 +41,7 @@ import kaldifeat import kaldifst import torch import torchaudio -from kaldi_hmm_gmm import DecodableCtc, FasterDecoder, FasterDecoderOptions +from kaldi_decoder import DecodableCtc, FasterDecoder, FasterDecoderOptions from torch.nn.utils.rnn import pad_sequence diff --git a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py index d1e1c4af2..315db2541 100755 --- a/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py +++ b/egs/librispeech/ASR/conformer_ctc/jit_pretrained_decode_with_HLG.py @@ -40,7 +40,7 @@ import kaldifeat import kaldifst import torch import torchaudio -from kaldi_hmm_gmm import DecodableCtc, FasterDecoder, FasterDecoderOptions +from kaldi_decoder import DecodableCtc, FasterDecoder, FasterDecoderOptions from torch.nn.utils.rnn import pad_sequence diff --git a/requirements.txt b/requirements.txt index c031d683c..5a8326619 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ kaldifst kaldilm kaldialign -kaldi-hmm-gmm +kaldi-decoder sentencepiece>=0.1.96 tensorboard typeguard