From 4e05d70f457f7afc3151280905045c0046c3127d Mon Sep 17 00:00:00 2001 From: Bailey Hirota Date: Wed, 6 Aug 2025 11:08:19 +0900 Subject: [PATCH] fix stash commit --- egs/mls_english/ASR/zipformer/train.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/egs/mls_english/ASR/zipformer/train.py b/egs/mls_english/ASR/zipformer/train.py index cdc4bdad3..2bd51ec49 100755 --- a/egs/mls_english/ASR/zipformer/train.py +++ b/egs/mls_english/ASR/zipformer/train.py @@ -1216,7 +1216,8 @@ def run(rank, world_size, args): return True 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: @@ -1237,10 +1238,8 @@ def run(rank, world_size, args): else: cuts_musan = None - - train_dl = mls_english_corpus.train_dataloader( - sampler_state_dict=sampler_state_dict, - cuts_musan=cuts_musan, + train_dl = mls_english_corpus.train_dataloaders( + train_cuts, sampler_state_dict=sampler_state_dict ) valid_dl = mls_english_corpus.valid_dataloader()