iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #98 from liulu112601/main

Browse Source 
Two Minor Revisions
This commit is contained in:
D-X-Y 2021-04-29 17:19:18 +08:00 committed by GitHub
commit d423ca9b20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/models/xcore.py
View File

@ -1,5 +1,5 @@
##################################################### #######################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 # # Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2021.04 #
####################################################### #######################################################
# Use module in xlayers to construct different models # # Use module in xlayers to construct different models #
####################################################### #######################################################

1
tests/test_super_container.py
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(),
) )