mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

add unit test for sep_fcn_head: debug 1

Browse Source
This commit is contained in:
johnzja 2020-08-12 10:20:47 +08:00
parent 36228f54dc
commit 2162e7f11f
1 changed files with 2 additions and 1 deletions

3
tests/test_models/test_heads.py

@ -549,7 +549,8 @@ def test_sep_fcn_head():
channels=128, channels=128,
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))
x = torch.rand(1, 128, 32, 32) x = torch.rand(1, 128, 32, 32)
output = head(x) output = head(x)
assert output.shape == (1, head.num_classes, 32, 32) assert output.shape == (1, head.num_classes, 32, 32)