wip

Author: ChadLin9596

src/py_utils/utils_torch.py

@@ -44,7 +44,7 @@ def seed_everything(seed=42, verbose=False):
     torch.cuda.manual_seed_all(seed)  # if you are using multi-GPU.
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark = False
-    torch.use_deterministic_algorithms(True)
+    torch.use_deterministic_algorithms(True, warn_only=True)
 
     if verbose:
         print(f"Seed set to: {seed} (type: {type(seed)})")