From 4cc0b090536a2deb46f85846a2b46d30ef639336 Mon Sep 17 00:00:00 2001 From: juliustao Date: Sun, 23 Jan 2022 19:26:16 -0500 Subject: [PATCH] use loader seed to set numba random state inside thread --- ffcv/loader/epoch_iterator.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ffcv/loader/epoch_iterator.py b/ffcv/loader/epoch_iterator.py index 4bc1b9bd..b6d4a5be 100644 --- a/ffcv/loader/epoch_iterator.py +++ b/ffcv/loader/epoch_iterator.py @@ -4,6 +4,7 @@ from contextlib import nullcontext from typing import Sequence, TYPE_CHECKING +import numpy as np import torch as ch from ..traversal_order.quasi_random import QuasiRandom @@ -66,6 +67,7 @@ def __init__(self, loader: 'Loader', order: Sequence[int]): self.start() def run(self): + Compiler.compile(lambda seed: np.random.seed(seed))(self.loader.seed) events = [None for _ in self.cuda_streams]