Reformulate Math Functions

2021-04-26 05:16:38 -07:00
parent 1980779053
commit e1818694a4
14 changed files with 308 additions and 254 deletions
--- a/tests/test_super_container.py
+++ b/tests/test_super_container.py
@@ -79,7 +79,7 @@ def test_super_sequential_v1():
        super_core.SuperSimpleNorm(1, 1),
        torch.nn.ReLU(),
        super_core.SuperLinear(10, 10),
-        super_core.SuperReLU()
+        super_core.SuperReLU(),
    )
    inputs = torch.rand(10, 10)
    print(model)