diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.asr_datamodule.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.asr_datamodule.py.swp index 0dcaa319c..8946716b3 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.asr_datamodule.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.asr_datamodule.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/asr_datamodule.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/asr_datamodule.py index 4fa3782a3..65613536a 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/asr_datamodule.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/asr_datamodule.py @@ -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" + ) +