Change back to changing all seeds

jonasburian · Oct 29, 2024 · 269e1a1 · 269e1a1
1 parent b3535ec
commit 269e1a1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/bulkmodel.py b/bulkmodel.py
@@ -34,9 +34,9 @@ def run_main(args):
     torch.cuda.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)
     #from transformers import *
-    random.seed(42)
-    np.random.seed(42)
-    os.environ['PYTHONHASHSEED'] = str(42)
+    random.seed(seed)
+    np.random.seed(seed)
+    os.environ['PYTHONHASHSEED'] = str(seed)
     #torch.manual_seed(seed)
     #torch.cuda.manual_seed(seed)
     torch.backends.cudnn.deterministic = True

diff --git a/scmodel.py b/scmodel.py
@@ -57,9 +57,9 @@ def run_main(args):
     torch.cuda.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)
     #from transformers import *
-    random.seed(42)
-    np.random.seed(42)
-    os.environ['PYTHONHASHSEED'] = str(42)
+    random.seed(seed)
+    np.random.seed(seed)
+    os.environ['PYTHONHASHSEED'] = str(seed)
 
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark=False