fix pd

deepmodeling · Dec 19, 2024 · 63dc3d3 · 63dc3d3
1 parent cf6e6e2
commit 63dc3d3
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/deepmd/pd/utils/dataloader.py b/deepmd/pd/utils/dataloader.py
@@ -30,12 +30,16 @@
     default_collate_fn,
 )
 
+from deepmd.utils import dp_random
 from deepmd.pd.utils import (
     env,
 )
 from deepmd.pd.utils.dataset import (
     DeepmdDataSetForLoader,
 )
+from deepmd.pt.utils.utils import (
+    mix_entropy,
+)
 from deepmd.utils.data import (
     DataRequirementItem,
 )
@@ -50,8 +54,13 @@
 
 
 def setup_seed(seed):
-    paddle.seed(seed)
+    if isinstance(seed, (list, tuple)):
+        mixed_seed = mix_entropy(seed)
+    else:
+        mixed_seed = seed
+    paddle.seed(mixed_seed)
     os.environ["FLAGS_cudnn_deterministic"] = "True"
+    dp_random.seed(seed)
 
 
 class DpLoaderSet(Dataset):