diff --git a/egs/multi_ja_en/ASR/zipformer/train.py b/egs/multi_ja_en/ASR/zipformer/train.py index c05d5edc3..890228ddd 100755 --- a/egs/multi_ja_en/ASR/zipformer/train.py +++ b/egs/multi_ja_en/ASR/zipformer/train.py @@ -1178,7 +1178,7 @@ def run(rank, world_size, args): if params.inf_check: register_inf_check_hooks(model) - reazonspeech_corpus = ReazonSpeechAsrDataModule(args) + multidataset_datamodule = MultiDatasetAsrDataModule(args) multi_dataset = MultiDataset(args) @@ -1244,7 +1244,7 @@ def run(rank, world_size, args): ) valid_cuts = multi_dataset.dev_cuts() - valid_dl = reazonspeech_corpus.valid_dataloaders(valid_cuts) + valid_dl = multidataset_datamodule.valid_dataloaders(valid_cuts) if not params.print_diagnostics: scan_pessimistic_batches_for_oom( @@ -1386,7 +1386,7 @@ def scan_pessimistic_batches_for_oom( def main(): parser = get_parser() - ReazonSpeechAsrDataModule.add_arguments(parser) + MultiDatasetAsrDataModule.add_arguments(parser) args = parser.parse_args() args.exp_dir = Path(args.exp_dir)