Update _core.py

interTwin-eu · Dec 12, 2024 · 5d5c2ad · 5d5c2ad
1 parent fb9b30b
commit 5d5c2ad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/use-cases/lattice-qcd/src/device/_core.py b/use-cases/lattice-qcd/src/device/_core.py
@@ -36,6 +36,7 @@ def __init__(self, model):
     def set_seed(self, seed):
         self.seed = seed
         torch.manual_seed(self.seed)
+        print(f"Seed set for rank {self.rank}: {self.seed}")
 
     def to(self, *args, **kwargs):
         self._model.net_.to(*args, **kwargs)