From 2162e7f11ff39aaf029f741993330c59892f1d18 Mon Sep 17 00:00:00 2001 From: johnzja Date: Wed, 12 Aug 2020 10:20:47 +0800 Subject: [PATCH] add unit test for sep_fcn_head: debug 1 --- tests/test_models/test_heads.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_models/test_heads.py b/tests/test_models/test_heads.py index c3a6ed647..8182e28a5 100644 --- 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)