Making torch deterministic for the tests

johnmarktaylor91 · Aug 2, 2024 · 222753a · 222753a
1 parent 89b5bce
commit 222753a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_validation_and_visuals.py b/tests/test_validation_and_visuals.py
@@ -30,7 +30,7 @@
 # Define inputs
 torch.manual_seed(0)
 torch.cuda.manual_seed_all(0)
-torch.use_deterministic_algorithms()
+torch.use_deterministic_algorithms(True)
 
 sub_dirs = [
     "cornet",