From 2965e3403828fc0e08875cc43416bdc49c313609 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 27 Oct 2021 15:07:14 +0100 Subject: [PATCH 1/2] setting 100 seeds --- test/test_transforms_tensor.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/test_transforms_tensor.py b/test/test_transforms_tensor.py index 5ee1c738a77..c6fcde2b4c0 100644 --- a/test/test_transforms_tensor.py +++ b/test/test_transforms_tensor.py @@ -115,9 +115,15 @@ def test_random(func, method, device, channels, fn_kwargs, match_kwargs): _test_op(func, method, device, channels, fn_kwargs, fn_kwargs, **match_kwargs) +# @pytest.mark.parametrize("seed", range(100)) +@pytest.mark.parametrize("seed", range(100)) @pytest.mark.parametrize("device", cpu_and_gpu()) @pytest.mark.parametrize("channels", [1, 3]) class TestColorJitter: + @pytest.fixture(autouse=True) + def set_random_seed(self, seed): + torch.random.manual_seed(seed) + @pytest.mark.parametrize("brightness", [0.1, 0.5, 1.0, 1.34, (0.3, 0.7), [0.4, 0.5]]) def test_color_jitter_brightness(self, brightness, device, channels): tol = 1.0 + 1e-10 From d1dc69cabfe4a1c986000cd59fc63735d5b1b70a Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Wed, 27 Oct 2021 16:38:39 +0100 Subject: [PATCH 2/2] back to 10 seeds --- test/test_transforms_tensor.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/test/test_transforms_tensor.py b/test/test_transforms_tensor.py index c6fcde2b4c0..df180600225 100644 --- a/test/test_transforms_tensor.py +++ b/test/test_transforms_tensor.py @@ -115,8 +115,7 @@ def test_random(func, method, device, channels, fn_kwargs, match_kwargs): _test_op(func, method, device, channels, fn_kwargs, fn_kwargs, **match_kwargs) -# @pytest.mark.parametrize("seed", range(100)) -@pytest.mark.parametrize("seed", range(100)) +@pytest.mark.parametrize("seed", range(10)) @pytest.mark.parametrize("device", cpu_and_gpu()) @pytest.mark.parametrize("channels", [1, 3]) class TestColorJitter: