Fix/patch how fix_random_seed() is imported.

This commit is contained in:
Daniel Povey 2022-03-26 15:43:47 +08:00
parent 26a1730392
commit 8a38d9a855

View File

@ -23,7 +23,7 @@ from functools import lru_cache
from pathlib import Path
from typing import Any, Dict, Optional
import torch
import lhotse
from lhotse.utils import fix_random_seed
from lhotse import CutSet, Fbank, FbankConfig, load_manifest
from lhotse.dataset import (
@ -307,7 +307,7 @@ class LibriSpeechAsrDataModule:
# set in the main process.
seed = torch.randint(0, 100000, ()).item()
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,