ADD file via upload

2024-08-03 00:45:53 +08:00 · 2024-08-03 00:45:53 +08:00 · 6da390e100
parent e76fc4ec07
commit 6da390e100
1 changed files with 14 additions and 0 deletions
--- a/set_seed.py
+++ b/set_seed.py
@ -0,0 +1,14 @@
+import random
+import numpy as np
+import torch
+
+
+def set_random_seed(seed):
+    """
+        set random seed
+    """
+    random.seed(seed)
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)