mirror of https://github.com/open-mmlab/mmcv.git
17 lines
420 B
Python
17 lines
420 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import torch
|
|
import torch.nn.functional as F
|
|
|
|
from mmcv.cnn.bricks import Swish
|
|
|
|
|
|
def test_swish():
|
|
act = Swish()
|
|
input = torch.randn(1, 3, 64, 64)
|
|
expected_output = input * F.sigmoid(input)
|
|
output = act(input)
|
|
# test output shape
|
|
assert output.shape == expected_output.shape
|
|
# test output value
|
|
assert torch.equal(output, expected_output)
|