"src/diffusers/schedulers/scheduling_sde_ve_flax.py" did not exist on "63c68d979a8e4a9c9fea306bdf63e73812843453"
test_swish.py 420 Bytes
Newer Older
limm's avatar
limm committed
1
# Copyright (c) OpenMMLab. All rights reserved.
Xiaojie Li's avatar
Xiaojie Li committed
2
import torch
limm's avatar
limm committed
3
import torch.nn.functional as F
Xiaojie Li's avatar
Xiaojie Li committed
4
5
6
7
8
9
10

from mmcv.cnn.bricks import Swish


def test_swish():
    act = Swish()
    input = torch.randn(1, 3, 64, 64)
limm's avatar
limm committed
11
    expected_output = input * F.sigmoid(input)
Xiaojie Li's avatar
Xiaojie Li committed
12
13
14
15
16
    output = act(input)
    # test output shape
    assert output.shape == expected_output.shape
    # test output value
    assert torch.equal(output, expected_output)