mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Manually fix merge conflict in multi_ja_en/ASR/zipformer/train.py
This commit is contained in:
parent
636121c507
commit
0967f5f7d1
@ -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)
|
||||||
|
|
||||||
@ -1244,7 +1244,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(
|
||||||
@ -1386,7 +1386,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)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user