pytest "set_deterministic"

ceb1a627 · panning · 3c106aa4 · ceb1a627
Commit ceb1a627 authored Nov 14, 2022 by panning
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

test/test_transforms_tensor.py test/test_transforms_tensor.py +3 -1

No files found.
--- a/test/test_transforms_tensor.py
+++ b/test/test_transforms_tensor.py
@@ -676,7 +676,9 @@ class Tester(TransformsTester):
 class CUDATester(Tester):

    def setUp(self):
-        torch.set_deterministic(False)
+        #pan
+        #torch.set_deterministic(False)
+        torch.use_deterministic_algorithms(False)
        self.device = "cuda"