Setting seeds for frozen BN. (#4761)

5ffba76f · Vasilis Vryniotis · GitHub · 8a62d542 · 5ffba76f
Unverified Commit 5ffba76f authored Oct 27, 2021 by Vasilis Vryniotis Committed by GitHub Oct 27, 2021
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

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

No files found.
--- a/test/test_ops.py
+++ b/test/test_ops.py
@@ -845,7 +845,9 @@ class TestFrozenBNT:
        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(