mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
add unit test for sep_fcn_head: debug 4
This commit is contained in:
parent
03ca449baa
commit
c89674d6cb
@ -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)
|
||||
|
Loading…
x
Reference in New Issue
Block a user