Manually fix merge conflict in multi_ja_en/ASR/zipformer/train.py

This commit is contained in:
Kinan Martin 2025-07-28 17:59:47 +09:00
parent aed139f125
commit dbd89773d5

View File

@ -1178,7 +1178,7 @@ def run(rank, world_size, args):
if params.inf_check: if params.inf_check:
register_inf_check_hooks(model) register_inf_check_hooks(model)
reazonspeech_corpus = ReazonSpeechAsrDataModule(args) multidataset_datamodule = MultiDatasetAsrDataModule(args)
multi_dataset = MultiDataset(args) multi_dataset = MultiDataset(args)
@ -1241,7 +1241,7 @@ def run(rank, world_size, args):
) )
valid_cuts = multi_dataset.dev_cuts() 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: if not params.print_diagnostics:
scan_pessimistic_batches_for_oom( scan_pessimistic_batches_for_oom(
@ -1383,7 +1383,7 @@ def scan_pessimistic_batches_for_oom(
def main(): def main():
parser = get_parser() parser = get_parser()
ReazonSpeechAsrDataModule.add_arguments(parser) MultiDatasetAsrDataModule.add_arguments(parser)
args = parser.parse_args() args = parser.parse_args()
args.exp_dir = Path(args.exp_dir) args.exp_dir = Path(args.exp_dir)