We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 297e2b8 commit fc63f82Copy full SHA for fc63f82
test/test_models.py
@@ -406,7 +406,7 @@ def test_mobilenet_norm_layer(model_fn):
406
assert any(isinstance(x, nn.BatchNorm2d) for x in model.modules())
407
408
def get_gn(num_channels):
409
- return nn.GroupNorm(32, num_channels)
+ return nn.GroupNorm(1, num_channels)
410
411
model = model_fn(norm_layer=get_gn)
412
assert not (any(isinstance(x, nn.BatchNorm2d) for x in model.modules()))
0 commit comments