[fbsync] Fix seed for TestColorJitter (#4762)

Summary:
* setting 100 seeds

* back to 10 seeds

Reviewed By: datumbox

Differential Revision: D32064707

fbshipit-source-id: 4cbe0883257f949b146c08f627c51e4160a14cda

Author: fmassa

test/test_transforms_tensor.py

@@ -115,9 +115,14 @@ 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(10))
 @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