[fbsync] Setting seeds for frozen BN. (#4761)

Reviewed By: datumbox

Differential Revision: D32064702

fbshipit-source-id: 9f7f6d3ffb346b85d34052326d6b5d1ad1ebba90

Author: fmassa

test/test_ops.py

@@ -845,7 +845,9 @@ def test_frozenbatchnorm2d_repr(self):
         expected_string = f"FrozenBatchNorm2d({num_features}, eps={eps})"
         assert repr(t) == expected_string
 
-    def test_frozenbatchnorm2d_eps(self):
+    @pytest.mark.parametrize("seed", range(10))
+    def test_frozenbatchnorm2d_eps(self, seed):
+        torch.random.manual_seed(seed)
         sample_size = (4, 32, 28, 28)
         x = torch.rand(sample_size)
         state_dict = dict(