Update weight watcher codes

2020-07-05 22:29:26 +00:00
parent 9659f132be
commit 6facc39a42
9 changed files with 167 additions and 161 deletions
--- a/lib/nas_201_api/api_201.py
+++ b/lib/nas_201_api/api_201.py
@@ -77,6 +77,7 @@ class NASBench201API(NASBenchMetaAPI):
    self.meta_archs = copy.deepcopy( file_path_or_dict['meta_archs'] )
    # This is a dict mapping each architecture to a dict, where the key is #epochs and the value is ArchResults
    self.arch2infos_dict = OrderedDict()
+    self._avaliable_hps = set(['12', '200'])
    for xkey in sorted(list(file_path_or_dict['arch2infos'].keys())):
      all_info = file_path_or_dict['arch2infos'][xkey]
      hp2archres = OrderedDict()