Turn on deterministic for histogram.

pytorch · Dec 4, 2020 · d84cb56 · d84cb56
1 parent 84fc498
commit d84cb56
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/test/test_functional_tensor.py b/test/test_functional_tensor.py
@@ -923,6 +923,7 @@ def test_autocontrast(self):
         )
 
     def test_equalize(self):
+        torch.set_deterministic(False)
         self._test_adjust_fn(
             F.equalize,
             F_pil.equalize,

diff --git a/test/test_transforms_tensor.py b/test/test_transforms_tensor.py
@@ -108,6 +108,7 @@ def test_random_autocontrast(self):
         self._test_op('autocontrast', 'RandomAutocontrast')
 
     def test_random_equalize(self):
+        torch.set_deterministic(False)
         self._test_op('equalize', 'RandomEqualize')
 
     def test_color_jitter(self):