Fix multi-process dataloader with same random seeds during epochs

PaddlePaddle · Jul 9, 2021 · 0adef44 · 0adef44
1 parent 1798a3e
commit 0adef44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddleseg/utils/utils.py b/paddleseg/utils/utils.py
@@ -14,10 +14,10 @@
 
 import contextlib
 import filelock
-import math
 import os
 import tempfile
 import numpy as np
+import random
 from urllib.parse import urlparse, unquote
 
 import paddle
@@ -122,4 +122,4 @@ def resume(model, optimizer, resume_model):
 
 
 def worker_init_fn(worker_id):
-    np.random.seed(np.random.get_state()[1][0] + worker_id)
+    np.random.seed(random.randint(0, 100000))