from local

This commit is contained in:
dohe0342 2023-01-03 13:06:18 +09:00
parent d7dbd21375
commit 45a138618b
2 changed files with 11 additions and 0 deletions

View File

@ -496,3 +496,14 @@ class LibriSpeechAsrDataModule:
return load_manifest_lazy(
self.args.manifest_dir / f"librispeech_cuts_test-other_{option}.jsonl"
)
elif option == 'user':
json_list = sorted(glob(self.args.manifest_dir / "/userlibri/test-other/*"))
spk_list = [json.split('/')[-1][:-6] for json in json_list]
return [load_manifest_lazy(json) for json in json_list], spk_list
else:
return load_manifest_lazy(
self.args.manifest_dir / f"librispeech_cuts_test-clean_{option}.jsonl"
)