diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp index 9f24745a4..9bd6d1f3d 100644 Binary files a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp differ diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode_old.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode_old.py.swp index 80e255019..2fa681695 100644 Binary files a/egs/librispeech/ASR/conformer_ctc2/.decode_old.py.swp and b/egs/librispeech/ASR/conformer_ctc2/.decode_old.py.swp differ diff --git a/egs/librispeech/ASR/conformer_ctc2/decode.py b/egs/librispeech/ASR/conformer_ctc2/decode.py index ceb129541..e8a211fc9 100755 --- a/egs/librispeech/ASR/conformer_ctc2/decode.py +++ b/egs/librispeech/ASR/conformer_ctc2/decode.py @@ -852,6 +852,7 @@ def main() -> None: logging.info(f"Number of model parameters: {num_param}") # we need cut ids to display recognition results. + ''' args.return_cuts = True tedlium = TedLiumAsrDataModule(args) @@ -865,6 +866,18 @@ def main() -> None: test_dls = [valid_dl, test_dl] #test_sets = ["dev"] #test_dls = [valid_dl] + ''' + args.return_cuts = True + librispeech = LibriSpeechAsrDataModule(args) + + test_clean_cuts = librispeech.test_clean_cuts() + test_other_cuts = librispeech.test_other_cuts() + + test_clean_dl = librispeech.test_dataloaders(test_clean_cuts) + test_other_dl = librispeech.test_dataloaders(test_other_cuts) + + test_sets = ["test-clean", "test-other"] + test_dl = [test_clean_dl, test_other_dl] for test_set, test_dl in zip(test_sets, test_dls): results_dict = decode_dataset(