Skip to content

GitLab

Unverified Commit fc63f828 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix mobilnet norm layer test (#5643) 

* xfail mobilnet norm layer test

* fix test
parent 297e2b87
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_models.py
View file @ fc63f828
...@@ -406,7 +406,7 @@ def test_mobilenet_norm_layer(model_fn): ...@@ -406,7 +406,7 @@ def test_mobilenet_norm_layer(model_fn):
assert any(isinstance(x, nn.BatchNorm2d) for x in model.modules()) assert any(isinstance(x, nn.BatchNorm2d) for x in model.modules())
def get_gn(num_channels): def get_gn(num_channels):
return nn.GroupNorm(32, num_channels) return nn.GroupNorm(1, num_channels)
model = model_fn(norm_layer=get_gn) model = model_fn(norm_layer=get_gn)
assert not (any(isinstance(x, nn.BatchNorm2d) for x in model.modules())) assert not (any(isinstance(x, nn.BatchNorm2d) for x in model.modules()))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment