mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-11 02:52:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

support for cascades

Browse Source
This commit is contained in:
yifanyeung 2024-07-04 16:09:20 +08:00
parent 91488ce972
commit 2502987dc4
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
egs/librispeech/ASR/zipformer_lstm/train.py
View File

@ -1262,12 +1262,19 @@ def run(rank, world_size, args):
sampler_state_dict = None sampler_state_dict = None
train_dl = librispeech.train_dataloaders( train_dl = librispeech.train_dataloaders(
train_cuts, sampler_state_dict=sampler_state_dict train_cuts,
sampler_state_dict=sampler_state_dict,
world_size=world_size,
rank=rank,
) )
valid_cuts = librispeech.dev_clean_cuts() valid_cuts = librispeech.dev_clean_cuts()
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,
world_size=world_size,
rank=rank,
)
if 0 and not params.print_diagnostics: if 0 and not params.print_diagnostics:
scan_pessimistic_batches_for_oom( scan_pessimistic_batches_for_oom(