Ejemplo n.º 1
0
def test_activations_init(activation):
    act = modules.activation_from_name(activation)
    res = act(INP)
    assert res.mean()
Ejemplo n.º 2
0
def test_activations_init(activation):
    inp = torch.ones(10)
    act = modules.activation_from_name(activation)
    res = act(inp)
    assert res.mean()