Update eval.py

BioinfoMachineLearning · Feb 21, 2025 · 78b82da · 78b82da
1 parent 0a7838f
commit 78b82da
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/flowdock/eval.py b/flowdock/eval.py
@@ -1,6 +1,7 @@
 import os
 
 import hydra
+import lightning as L
 import lovely_tensors as lt
 import rootutils
 import torch
@@ -56,6 +57,10 @@ def evaluate(cfg: DictConfig) -> Tuple[Dict[str, Any], Dict[str, Any]]:
     assert cfg.ckpt_path, "Please provide a checkpoint path to evaluate!"
     assert os.path.exists(cfg.ckpt_path), f"Checkpoint path {cfg.ckpt_path} does not exist!"
 
+    # set seed for random number generators in pytorch, numpy and python.random
+    if cfg.get("seed"):
+        L.seed_everything(cfg.seed, workers=True)
+
     log.info(
         f"Setting `float32_matmul_precision` to {cfg.model.cfg.task.float32_matmul_precision}."
     )