mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
add unit test for inverted_residual: debug 1
This commit is contained in:
parent
0d0641b5c1
commit
b3bd282cc7
@ -10,7 +10,7 @@ def test_inv_residual():
|
||||
# test stride assertion.
|
||||
InvertedResidual(32, 32, 3, 4)
|
||||
|
||||
# test default config with res connection.
|
||||
# test default config without res connection.
|
||||
# set expand_ratio = 4, stride = 1 and inp=oup.
|
||||
inv_module = InvertedResidual(32, 32, 1, 4)
|
||||
assert inv_module.use_res_connect
|
||||
@ -18,8 +18,8 @@ def test_inv_residual():
|
||||
assert inv_module.conv[0].padding == 0
|
||||
assert inv_module.conv[1].kernel_size == (3, 3)
|
||||
assert inv_module.conv[1].padding == 1
|
||||
assert not inv_module.conv[0].with_norm
|
||||
assert not inv_module.conv[1].with_norm
|
||||
assert inv_module.conv[0].with_norm
|
||||
assert inv_module.conv[1].with_norm
|
||||
x = torch.rand(1, 32, 64, 64)
|
||||
output = inv_module(x)
|
||||
assert output.shape == (1, 32, 64, 64)
|
||||
|
Loading…
x
Reference in New Issue
Block a user