Merge branch 'main' into malfet/prefer-nvidia-channel-for-conda-builds

pytorch · Mar 20, 2022 · 33a72da · 33a72da
2 parents 2366633 + fc63f82
commit 33a72da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()))