add unit test for sep_fcn_head: debug 2

pull/58/head
johnzja 2020-08-12 10:26:07 +08:00
parent 2162e7f11f
commit 0466908ff2
1 changed files with 2 additions and 2 deletions

View File

@ -551,9 +551,9 @@ def test_sep_fcn_head():
num_classes=19,
in_index=-1,
norm_cfg=dict(type='SyncBN', requires_grad=True, momentum=0.01))
x = torch.rand(1, 128, 32, 32)
x = torch.rand(2, 128, 32, 32)
output = head(x)
assert output.shape == (1, head.num_classes, 32, 32)
assert output.shape == (2, head.num_classes, 32, 32)
assert not head.concat_input
from mmseg.ops.separable_conv_module import DepthwiseSeparableConvModule
assert isinstance(head.convs[0], DepthwiseSeparableConvModule)