From c89674d6cb951638e8086b38252a12fd1f9e448b Mon Sep 17 00:00:00 2001 From: johnzja Date: Wed, 12 Aug 2020 10:29:15 +0800 Subject: [PATCH] add unit test for sep_fcn_head: debug 4 --- tests/test_models/test_heads.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models/test_heads.py b/tests/test_models/test_heads.py index b49ecd018..dc41e410b 100644 --- a/tests/test_models/test_heads.py +++ b/tests/test_models/test_heads.py @@ -550,7 +550,7 @@ def test_sep_fcn_head(): concat_input=False, num_classes=19, in_index=-1, - norm_cfg=dict(type='SyncBN', requires_grad=True, momentum=0.01)) + norm_cfg=dict(type='BN', requires_grad=True, momentum=0.01)) x = [torch.rand(2, 128, 32, 32)] output = head(x) assert output.shape == (2, head.num_classes, 32, 32)