Update NATS-Bench (tss version 1.0) and remove the trace of 301

2020-09-16 08:28:27 +00:00
parent bd9288f45d
commit 9db28392c2
10 changed files with 169 additions and 249 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' or name == 'nats-bench' or name == 'nats-bench-size':
+    if name in ['nats-bench', 'nats-bench-size']:
      return {'candidates': [8, 16, 24, 32, 40, 48, 56, 64],
              'numbers': 5}
    else: