Set seed of test_random_erasing (#4764)

* Trying with 100 seeds * Change seeds from 100 to 10

Set seed of test_random_erasing (#4764)
* Trying with 100 seeds * Change seeds from 100 to 10
e00d8189 · Nicolas Hug · GitHub · 3a17e339 · e00d8189
Unverified Commit e00d8189 authored Oct 27, 2021 by Nicolas Hug Committed by GitHub Oct 27, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

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

No files found.
--- a/test/test_transforms.py
+++ b/test/test_transforms.py
@@ -1750,8 +1750,10 @@ def test_color_jitter():
    color_jitter.__repr__()


+@pytest.mark.parametrize("seed", range(10))
 @pytest.mark.skipif(stats is None, reason="scipy.stats not available")
-def test_random_erasing():
+def test_random_erasing(seed):
+    torch.random.manual_seed(seed)
    img = torch.ones(3, 128, 128)

    t = transforms.RandomErasing(scale=(0.1, 0.1), ratio=(1 / 3, 3.0))