From 44f0c8bd8abd738e2e22c36e717b6f6b313e20b7 Mon Sep 17 00:00:00 2001 From: jinzr Date: Thu, 7 Dec 2023 11:15:45 +0800 Subject: [PATCH] Update streaming_ctc_decode.py --- egs/librispeech/ASR/zipformer/streaming_ctc_decode.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/streaming_ctc_decode.py b/egs/librispeech/ASR/zipformer/streaming_ctc_decode.py index 525c0f405..3e7b5e1e8 100755 --- a/egs/librispeech/ASR/zipformer/streaming_ctc_decode.py +++ b/egs/librispeech/ASR/zipformer/streaming_ctc_decode.py @@ -631,8 +631,12 @@ def main(): librispeech = LibriSpeechAsrDataModule(args) - test_cuts = librispeech.test_cuts() - test_sets = {"test": test_cuts} + test_clean_cuts = librispeech.test_clean_cuts() + test_other_cuts = librispeech.test_other_cuts() + test_sets = { + "test-clean": test_clean_cuts, + "test-other": test_other_cuts, + } for test_set, test_cut in test_sets.items(): results_dict = decode_dataset(