From e234e4b9b032d3aad088c53c7b2d4d47f74023a5 Mon Sep 17 00:00:00 2001 From: sathvik udupa Date: Wed, 3 May 2023 11:06:14 +0530 Subject: [PATCH] small changes to decode --- egs/mucs/ASR/conformer_ctc/decode.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/egs/mucs/ASR/conformer_ctc/decode.py b/egs/mucs/ASR/conformer_ctc/decode.py index c5ac7912b..907fc6c8f 100755 --- a/egs/mucs/ASR/conformer_ctc/decode.py +++ b/egs/mucs/ASR/conformer_ctc/decode.py @@ -26,7 +26,7 @@ import k2 import sentencepiece as spm import torch import torch.nn as nn -from asr_datamodule import LibriSpeechAsrDataModule +from asr_datamodule import MUCSAsrDataModule from conformer import Conformer from icefall.bpe_graph_compiler import BpeCtcTrainingGraphCompiler @@ -618,7 +618,7 @@ def save_results( @torch.no_grad() def main(): parser = get_parser() - LibriSpeechAsrDataModule.add_arguments(parser) + MUCSAsrDataModule.add_arguments(parser) args = parser.parse_args() args.exp_dir = Path(args.exp_dir) args.lang_dir = Path(args.lang_dir) @@ -777,7 +777,7 @@ def main(): # we need cut ids to display recognition results. args.return_cuts = True - librispeech = LibriSpeechAsrDataModule(args) + librispeech = MUCSAsrDataModule(args) test_clean_cuts = librispeech.test_mucs_cuts()