diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.decode.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.decode.py.swp new file mode 100644 index 000000000..9d9a4349d Binary files /dev/null and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.decode.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/decode.py index 0df650264..5e820e796 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/decode.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/decode.py @@ -937,14 +937,18 @@ def main(): # we need cut ids to display recognition results. args.return_cuts = True librispeech = LibriSpeechAsrDataModule(args) - + + dev_clean_cuts = librispeech.dev_clean_cuts() + dev_other_cuts = librispeech.dev_other_cuts() test_clean_cuts = librispeech.test_clean_cuts() test_other_cuts = librispeech.test_other_cuts() - + + dev_clean_dl = librispeech.valid_dataloaders(dev_clean_cuts) + dev_other_dl = librispeech.valid_dataloaders(dev_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_sets = ["dev-clean", "dev-other", "test-clean", "test-other"] test_dl = [test_clean_dl, test_other_dl] for test_set, test_dl in zip(test_sets, test_dl):