def test_callable_deserialization(self): def dummy(x, **kwargs): return x for f in [dummy, lambda x: x]: f = activations.get_activation(f) iscallable = callable(f) self.assertTrue(iscallable) self.assertEqual(1, f(1))
def __init__(self, margin=0.0, squashing="identity", beta=10, **kwargs): super().__init__(**kwargs) self.margin = margin self.squashing = get_activation(squashing) self.beta = torch.tensor(beta)
def test_unknown_deserialization(self): for funcname in ["blubb", "foobar"]: with self.assertRaises(NameError): _ = activations.get_activation(funcname)
def test_funcname_deserialization(self): for funcname in self.flist: f = activations.get_activation(funcname) iscallable = callable(f) self.assertTrue(iscallable)