fix mobilnet norm layer test (#5643)

* xfail mobilnet norm layer test * fix test

fix mobilnet norm layer test (#5643)
* xfail mobilnet norm layer test * fix test
fc63f828 · Philip Meier · GitHub · 297e2b87 · fc63f828
Unverified Commit fc63f828 authored Mar 20, 2022 by Philip Meier Committed by GitHub Mar 20, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/test_models.py test/test_models.py +1 -1

No files found.
--- a/test/test_models.py
+++ b/test/test_models.py
@@ -406,7 +406,7 @@ def test_mobilenet_norm_layer(model_fn):
    assert any(isinstance(x, nn.BatchNorm2d) for x in model.modules())

    def get_gn(num_channels):
-        return nn.GroupNorm(32, num_channels)
+        return nn.GroupNorm(1, num_channels)

    model = model_fn(norm_layer=get_gn)
    assert not (any(isinstance(x, nn.BatchNorm2d) for x in model.modules()))