mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix stash commit

Browse Source
This commit is contained in:
Bailey Hirota 2025-08-06 11:08:19 +09:00
parent f9ceead59e
commit 4e05d70f45
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/mls_english/ASR/zipformer/train.py
View File

@ -1216,7 +1216,8 @@ def run(rank, world_size, args):
return True return True
mls_english_corpus = MLSEnglishHFAsrDataModule(args) mls_english_corpus = MLSEnglishHFAsrDataModule(args)
mls_english_corpus.load_dataset(args.dataset_path) train_cuts = mls_english_corpus.train_cuts()
# mls_english_corpus.load_dataset(args.dataset_path)
if params.start_batch > 0 and checkpoints and "sampler" in checkpoints: if params.start_batch > 0 and checkpoints and "sampler" in checkpoints:
@ -1237,10 +1238,8 @@ def run(rank, world_size, args):
else: else:
cuts_musan = None cuts_musan = None
train_dl = mls_english_corpus.train_dataloaders(
train_dl = mls_english_corpus.train_dataloader( train_cuts, sampler_state_dict=sampler_state_dict
sampler_state_dict=sampler_state_dict,
cuts_musan=cuts_musan,
) )
valid_dl = mls_english_corpus.valid_dataloader() valid_dl = mls_english_corpus.valid_dataloader()