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