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

PR review suggestions implemented

Browse Source
This commit is contained in:
Bailey Hirota 2025-07-17 02:01:03 +09:00 committed by Kinan Martin
parent 6012edbc17
commit dc4db379ea
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/multi_ja_en/ASR/local/utils/asr_datamodule.py
View File

@ -302,13 +302,14 @@ class MultiDatasetAsrDataModule:
logging.info("Loading sampler state dict") logging.info("Loading sampler state dict")
train_sampler.load_state_dict(sampler_state_dict) train_sampler.load_state_dict(sampler_state_dict)
seed = torch.randint(0, 100000, ()).item() seed = 42
worker_init_fn = _SeedWorkers(seed) worker_init_fn = _SeedWorkers(seed)
train_dl = DataLoader( train_dl = DataLoader(
train, train,
sampler=train_sampler, sampler=train_sampler,
batch_size=None, batch_size=None,
pin_memory=True,
num_workers=self.args.num_workers, num_workers=self.args.num_workers,
persistent_workers=True, persistent_workers=True,
worker_init_fn=worker_init_fn, worker_init_fn=worker_init_fn,

2
egs/multi_ja_en/ASR/local/utils/update_cutset_paths.py
View File

@ -46,7 +46,7 @@ def update_paths(cuts: CutSet, dataset_name: str, old_feature_prefix: str) -> Cu
logger.info( logger.info(
f"Updating cut {cut.id}: {original_storage_path} → {new_storage_path}" f"Updating cut {cut.id}: {original_storage_path} → {new_storage_path}"
) )
cut.features.storage_path = str(new_storage_path) new_storage_path.as_posix()
updated_cuts.append(cut) updated_cuts.append(cut)
else: else:
logger.warning(f"Skipping update for cut {cut.id}: has no features.") logger.warning(f"Skipping update for cut {cut.id}: has no features.")