From 255f9be2b2c1586abddfcd390bcd8314a3121a30 Mon Sep 17 00:00:00 2001 From: Yifan Yang Date: Wed, 14 Jun 2023 19:03:38 +0800 Subject: [PATCH] Fix train.py --- egs/multi_en/ASR/zipformer/train.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/egs/multi_en/ASR/zipformer/train.py b/egs/multi_en/ASR/zipformer/train.py index 756fe6e7e..7f5ff8ee7 100755 --- a/egs/multi_en/ASR/zipformer/train.py +++ b/egs/multi_en/ASR/zipformer/train.py @@ -31,7 +31,6 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3" --start-epoch 1 \ --use-fp16 1 \ --exp-dir zipformer/exp \ - --full-libri 1 \ --max-duration 1000 # For streaming model training: @@ -42,7 +41,6 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3" --use-fp16 1 \ --exp-dir zipformer/exp \ --causal 1 \ - --full-libri 1 \ --max-duration 1000 It supports training with: @@ -1178,7 +1176,7 @@ def run(rank, world_size, args): register_inf_check_hooks(model) librispeech = LibriSpeechAsrDataModule(args) - multidataset = MultiDataset(args) + multidataset = MultiDataset(args.manifest_dir) train_cuts = multidataset.train_cuts() @@ -1236,7 +1234,7 @@ def run(rank, world_size, args): valid_cuts += librispeech.dev_other_cuts() valid_dl = librispeech.valid_dataloaders(valid_cuts) - if not params.print_diagnostics: + if 0 and not params.print_diagnostics: scan_pessimistic_batches_for_oom( model=model, train_dl=train_dl,