mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add unit test for sep_fcn_head: debug 4

pull/58/head
johnzja 2020-08-12 10:29:15 +08:00
parent 03ca449baa
commit c89674d6cb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_models/test_heads.py
View File

@ -550,7 +550,7 @@ def test_sep_fcn_head():
concat_input=False, concat_input=False,
num_classes=19, num_classes=19,
in_index=-1, 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)] x = [torch.rand(2, 128, 32, 32)]
output = head(x) output = head(x)
assert output.shape == (2, head.num_classes, 32, 32) assert output.shape == (2, head.num_classes, 32, 32)