iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

beta-0.1

Browse Source
This commit is contained in:
D-X-Y
2019-11-08 20:06:12 +11:00
parent 5e44189d7e
commit 1da5b49018
9 changed files with 256 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/models/cell_operations.py
View File

@@ -17,7 +17,8 @@ CONNECT_NAS_BENCHMARK = ['none', 'skip_connect', 'nor_conv_3x3']
AA_NAS_BENCHMARK = ['none', 'skip_connect', 'nor_conv_1x1', 'nor_conv_3x3', 'avg_pool_3x3']
SearchSpaceNames = {'connect-nas' : CONNECT_NAS_BENCHMARK,
'aa-nas' : AA_NAS_BENCHMARK}
'aa-nas' : AA_NAS_BENCHMARK,
'full' : sorted(list(OPS.keys()))}
class ReLUConvBN(nn.Module):