mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix train.py

Browse Source
This commit is contained in:
Yifan Yang 2023-06-14 19:03:38 +08:00
parent 5b049a1a3a
commit 255f9be2b2
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/multi_en/ASR/zipformer/train.py
View File

@ -31,7 +31,6 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3"
--start-epoch 1 \ --start-epoch 1 \
--use-fp16 1 \ --use-fp16 1 \
--exp-dir zipformer/exp \ --exp-dir zipformer/exp \
--full-libri 1 \
--max-duration 1000 --max-duration 1000
# For streaming model training: # For streaming model training:
@ -42,7 +41,6 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3"
--use-fp16 1 \ --use-fp16 1 \
--exp-dir zipformer/exp \ --exp-dir zipformer/exp \
--causal 1 \ --causal 1 \
--full-libri 1 \
--max-duration 1000 --max-duration 1000
It supports training with: It supports training with:
@ -1178,7 +1176,7 @@ def run(rank, world_size, args):
register_inf_check_hooks(model) register_inf_check_hooks(model)
librispeech = LibriSpeechAsrDataModule(args) librispeech = LibriSpeechAsrDataModule(args)
multidataset = MultiDataset(args) multidataset = MultiDataset(args.manifest_dir)
train_cuts = multidataset.train_cuts() train_cuts = multidataset.train_cuts()
@ -1236,7 +1234,7 @@ def run(rank, world_size, args):
valid_cuts += librispeech.dev_other_cuts() valid_cuts += librispeech.dev_other_cuts()
valid_dl = librispeech.valid_dataloaders(valid_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( scan_pessimistic_batches_for_oom(
model=model, model=model,
train_dl=train_dl, train_dl=train_dl,