mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-09 14:05:33 +00:00
PR review suggestions implemented
This commit is contained in:
parent
6012edbc17
commit
dc4db379ea
@ -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,
|
||||
|
||||
@ -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.")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user