Apply 1 suggestion(s) to 1 file(s)

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)