Skip to content

Commit 31e016c

Browse files
committed
Turn off deterministic for histogram.
1 parent 84fc498 commit 31e016c

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

test/test_functional_tensor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -923,6 +923,7 @@ def test_autocontrast(self):
923923
)
924924

925925
def test_equalize(self):
926+
torch.set_deterministic(False)
926927
self._test_adjust_fn(
927928
F.equalize,
928929
F_pil.equalize,

test/test_transforms_tensor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -108,6 +108,7 @@ def test_random_autocontrast(self):
108108
self._test_op('autocontrast', 'RandomAutocontrast')
109109

110110
def test_random_equalize(self):
111+
torch.set_deterministic(False)
111112
self._test_op('equalize', 'RandomEqualize')
112113

113114
def test_color_jitter(self):

0 commit comments

Comments
 (0)