add unit test for sep_fcn_head: debug 1

2020-08-12 10:20:47 +08:00 · 2020-08-12 10:20:47 +08:00 · 2162e7f11f
parent 36228f54dc
commit 2162e7f11f
1 changed files with 2 additions and 1 deletions
--- a/tests/test_models/test_heads.py
+++ b/tests/test_models/test_heads.py
@ -549,7 +549,8 @@ def test_sep_fcn_head():
        channels=128,
        concat_input=False,
        num_classes=19,
-        in_index=-1)
+        in_index=-1,
+        norm_cfg=dict(type='SyncBN', requires_grad=True, momentum=0.01))
    x = torch.rand(1, 128, 32, 32)
    output = head(x)
    assert output.shape == (1, head.num_classes, 32, 32)