Update tests

This commit is contained in:
lu 2021-04-29 02:17:31 -07:00
parent 14905d0011
commit 5d84001817

View File

@ -78,6 +78,7 @@ def test_super_sequential_v1():
model = super_core.SuperSequential( model = super_core.SuperSequential(
super_core.SuperSimpleNorm(1, 1), super_core.SuperSimpleNorm(1, 1),
torch.nn.ReLU(), torch.nn.ReLU(),
super_core.SuperLeakyReLU(),
super_core.SuperLinear(10, 10), super_core.SuperLinear(10, 10),
super_core.SuperReLU(), super_core.SuperReLU(),
) )