From 222753a33e364588f83a483d414a5849c2a60cea Mon Sep 17 00:00:00 2001 From: JohnMark Taylor Date: Fri, 2 Aug 2024 14:47:42 -0400 Subject: [PATCH] Making torch deterministic for the tests --- tests/test_validation_and_visuals.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_validation_and_visuals.py b/tests/test_validation_and_visuals.py index 7181d2e..98de015 100644 --- 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",