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

Fix/patch how fix_random_seed() is imported.

Browse Source
This commit is contained in:
Daniel Povey 2022-03-26 15:43:47 +08:00
parent 26a1730392
commit 8a38d9a855
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/tdnn_lstm_ctc/asr_datamodule.py
View File

@ -23,7 +23,7 @@ from functools import lru_cache
from pathlib import Path from pathlib import Path
from typing import Any, Dict, Optional from typing import Any, Dict, Optional
import torch import torch
import lhotse from lhotse.utils import fix_random_seed
from lhotse import CutSet, Fbank, FbankConfig, load_manifest from lhotse import CutSet, Fbank, FbankConfig, load_manifest
from lhotse.dataset import ( from lhotse.dataset import (
@ -307,7 +307,7 @@ class LibriSpeechAsrDataModule:
# set in the main process. # set in the main process.
seed = torch.randint(0, 100000, ()).item() seed = torch.randint(0, 100000, ()).item()
def worker_init_fn(worker_id: int): def worker_init_fn(worker_id: int):
lhotse.utils.fix_random_seed(seed + worker_id) fix_random_seed(seed + worker_id)
train_dl = DataLoader( train_dl = DataLoader(
train, train,