Skip to content

Commit fc63f82

Browse files
authored
fix mobilnet norm layer test (#5643)
* xfail mobilnet norm layer test * fix test
1 parent 297e2b8 commit fc63f82

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -406,7 +406,7 @@ def test_mobilenet_norm_layer(model_fn):
406406
assert any(isinstance(x, nn.BatchNorm2d) for x in model.modules())
407407

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

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

0 commit comments

Comments
 (0)