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
parent 14905d0011 59d07b6342
commit d423ca9b20
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/models/xcore.py
View File

@@ -1,4 +1,4 @@
##################################################### #######################################################
# 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(),
) )