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/init.py
+++ b/lib/models/init.py
@@ -64,7 +64,7 @@ def get_search_spaces(xtype, name) -> List[Text]:
    assert name in SearchSpaceNames, 'invalid name [{:}] in {:}'.format(name, SearchSpaceNames.keys())
    return SearchSpaceNames[name]
  elif xtype == 'sss':  # The size search space.
-    if name == 'nas-bench-301':
+    if name == 'nas-bench-301' or name == 'nats-bench' or name == 'nats-bench-size':
      return {'candidates': [8, 16, 24, 32, 40, 48, 56, 64],
              'numbers': 5}
    else: