2022-02-24 09:24:25 +08:00
|
|
|
# Copyright (c) OpenMMLab. All rights reserved.
|
2020-06-28 23:15:47 +08:00
|
|
|
import torch
|
|
|
|
|
from torch.nn.functional import relu6
|
|
|
|
|
|
|
|
|
|
from mmcv.cnn.bricks import HSwish
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_hswish():
|
|
|
|
|
# test inplace
|
|
|
|
|
act = HSwish(inplace=True)
|
|
|
|
|
assert act.act.inplace
|
|
|
|
|
act = HSwish()
|
|
|
|
|
assert not act.act.inplace
|
|
|
|
|
|
|
|
|
|
input = torch.randn(1, 3, 64, 64)
|
|
|
|
|
expected_output = input * relu6(input + 3) / 6
|
|
|
|
|
output = act(input)
|
|
|
|
|
# test output shape
|
|
|
|
|
assert output.shape == expected_output.shape
|
|
|
|
|
# test output value
|
|
|
|
|
assert torch.equal(output, expected_output)
|
