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 3a896287d..8073ab7d1 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 2b73cc0db..c8a8d7a5c 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 @@ -398,7 +398,7 @@ class LibriSpeechAsrDataModule: @lru_cache() def train_clean_100_cuts(self, option=None) -> CutSet: logging.info("About to get train-clean-100 cuts") - if option is None + if option is None: return load_manifest_lazy( self.args.manifest_dir / f"librispeech_cuts_train-clean-100.jsonl" ) @@ -411,16 +411,27 @@ class LibriSpeechAsrDataModule: @lru_cache() def train_clean_360_cuts(self, option=None) -> CutSet: logging.info("About to get train-clean-360 cuts") - return load_manifest_lazy( - self.args.manifest_dir / f"librispeech_cuts_train-clean-360_{option}.jsonl" - ) + if option is None: + return load_manifest_lazy( + self.args.manifest_dir / f"librispeech_cuts_train-clean-360.jsonl" + ) + else: + return load_manifest_lazy( + self.args.manifest_dir / f"librispeech_cuts_train-clean-360_{option}.jsonl" + ) @lru_cache() def train_other_500_cuts(self, option=None) -> CutSet: logging.info("About to get train-other-500 cuts") - return load_manifest_lazy( - self.args.manifest_dir / f"librispeech_cuts_train-other-500_{option}.jsonl" - ) + if option is None: + return load_manifest_lazy( + self.args.manifest_dir / f"librispeech_cuts_train-other-500.jsonl" + ) + else: + return load_manifest_lazy( + self.args.manifest_dir / f"librispeech_cuts_train-other-500_{option}.jsonl" + ) + @lru_cache() def train_all_shuf_cuts(self, option=None) -> CutSet: