Create NATS

2020-07-30 13:07:11 +00:00
parent df45e68366
commit 6061d74631
21 changed files with 1336 additions and 126 deletions
--- a/lib/models/cell_operations.py
+++ b/lib/models/cell_operations.py
@@ -25,6 +25,7 @@ NAS_BENCH_201         = ['none', 'skip_connect', 'nor_conv_1x1', 'nor_conv_3x3',
 DARTS_SPACE           = ['none', 'skip_connect', 'dua_sepc_3x3', 'dua_sepc_5x5', 'dil_sepc_3x3', 'dil_sepc_5x5', 'avg_pool_3x3', 'max_pool_3x3']

 SearchSpaceNames = {'connect-nas'  : CONNECT_NAS_BENCHMARK,
+                    'nats-bench'   : NAS_BENCH_201,
                    'nas-bench-201': NAS_BENCH_201,
                    'nas-bench-301': NAS_BENCH_201,
                    'darts'        : DARTS_SPACE}