From f6325d9e50dabcc7f76fab1d9b733c2407a05279 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Thu, 8 Jun 2023 15:46:14 +0000
Subject: [PATCH] Apply 1 suggestion(s) to 1 file(s)
---
dan/manager/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dan/manager/dataset.py b/dan/manager/dataset.py
index c888543c..9d987c68 100644
--- a/dan/manager/dataset.py
+++ b/dan/manager/dataset.py
@@ -139,7 +139,7 @@ class DatasetManager:
)
@staticmethod
- def seed_worker():
+ def seed_worker(worker_id):
worker_seed = torch.initial_seed() % 2**32
np.random.seed(worker_seed)
random.seed(worker_seed)
--
GitLab