diff --git a/egs/multi_zh_en/ASR/zipformer/multi_dataset.py b/egs/multi_zh_en/ASR/zipformer/multi_dataset.py index 1bc514d61..c181c8af3 100644 --- a/egs/multi_zh_en/ASR/zipformer/multi_dataset.py +++ b/egs/multi_zh_en/ASR/zipformer/multi_dataset.py @@ -32,7 +32,7 @@ class MultiDataset: It is expected to contain the following files: - aishell2_cuts_train.jsonl.gz """ - self.fbank_dir = Path(args.fbank_dir) + self.fbank_dir = Path(args.manifest_dir) self.use_tal_csasr = args.use_tal_csasr def train_cuts(self) -> CutSet: diff --git a/egs/multi_zh_en/ASR/zipformer/train.py b/egs/multi_zh_en/ASR/zipformer/train.py index 0fcb7e51e..e1bf65da9 100755 --- a/egs/multi_zh_en/ASR/zipformer/train.py +++ b/egs/multi_zh_en/ASR/zipformer/train.py @@ -1180,7 +1180,7 @@ def run(rank, world_size, args): register_inf_check_hooks(model) data_module = AsrDataModule(args) - multi_dataset = MultiDataset(args.manifest_dir) + multi_dataset = MultiDataset(args) train_cuts = multi_dataset.train_cuts()