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

keep backward compatibility and add proper error handling for musan manifest loading

Browse Source
This commit is contained in:
Bailey Hirota 2025-07-11 19:07:45 +09:00
parent 345b2ab1b0
commit 7a56af4351
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/reazonspeech/ASR/local/utils/asr_datamodule.py
View File

@ -174,7 +174,7 @@ class ReazonSpeechAsrDataModule:
group.add_argument( group.add_argument(
"--enable-musan", "--enable-musan",
type=str2bool, type=str2bool,
default=True, default=False,
help="When enabled, select noise from MUSAN and mix it" help="When enabled, select noise from MUSAN and mix it"
"with training dataset. ", "with training dataset. ",
) )

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

@ -1221,9 +1221,12 @@ def run(rank, world_size, args):
sampler_state_dict = None sampler_state_dict = None
if args.enable_musan: if args.enable_musan:
cuts_musan = load_manifest(Path(args.manifest_dir) / "musan_cuts.jsonl.gz") musan_path = Path(args.manifest_dir) / "musan_cuts.jsonl.gz"
else: if musan_path.exists():
cuts_musan = None cuts_musan = load_manifest(musan_path)
logging.info(f"Loaded MUSAN manifest from {musan_path}")
else:
cuts_musan = None
train_dl = reazonspeech_corpus.train_dataloaders( train_dl = reazonspeech_corpus.train_dataloaders(
train_cuts, train_cuts,