From a567459179843cef6052fab9fc79ce9fc707fc0f Mon Sep 17 00:00:00 2001 From: sifaoso Date: Fri, 5 Jun 2026 16:31:18 +0200 Subject: [PATCH] [FIX] deterministic augmentations in MTA by setting seed of all used RNGs in the MTA multiprocessing environment --- batchgenerators/dataloading/multi_threaded_augmenter.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/batchgenerators/dataloading/multi_threaded_augmenter.py b/batchgenerators/dataloading/multi_threaded_augmenter.py index f55b823..35653ee 100755 --- a/batchgenerators/dataloading/multi_threaded_augmenter.py +++ b/batchgenerators/dataloading/multi_threaded_augmenter.py @@ -24,6 +24,7 @@ from multiprocessing import Event from time import sleep, time from threadpoolctl import threadpool_limits +import random try: import torch @@ -34,6 +35,8 @@ def producer(queue, data_loader, transform, thread_id, seed, abort_event, pause_event=None, wait_time: float = 0.02): np.random.seed(seed) + random.seed(seed) if seed is not None else None + torch.manual_seed(seed) if seed is not None else None data_loader.set_thread_id(thread_id) item = None