PR review suggestions implemented

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

View File

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

View File

@ -46,7 +46,7 @@ def update_paths(cuts: CutSet, dataset_name: str, old_feature_prefix: str) -> Cu
logger.info(
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)
else:
logger.warning(f"Skipping update for cut {cut.id}: has no features.")