Python activation_from_nameの例

コード例 #1

ファイル: test_modules.py プロジェクト: iampakos/pytorch-tools

def test_activations_init(activation):
    act = modules.activation_from_name(activation)
    res = act(INP)
    assert res.mean()

コード例 #2

def test_activations_init(activation):
    inp = torch.ones(10)
    act = modules.activation_from_name(activation)
    res = act(inp)
    assert res.mean()