from local

This commit is contained in:
dohe0342 2023-02-17 10:25:37 +09:00
parent 592e94980e
commit ae45d9178a
3 changed files with 13 additions and 0 deletions

View File

@ -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(