diff --git a/egs/multi_ja_en/ASR/zipformer/train.py b/egs/multi_ja_en/ASR/zipformer/train.py index d840db008..c4aaa17db 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) @@ -1241,7 +1241,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( @@ -1383,7 +1383,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)