from local

This commit is contained in:
dohe0342 2023-02-27 14:54:18 +09:00
parent 27bd3e5085
commit 110344a8e5
2 changed files with 7 additions and 3 deletions

View File

@ -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):