Update the query_by_arch function in API to be compatiable with the submission version of NAS-Bench-201

This commit is contained in:
D-X-Y
2020-07-08 05:08:55 +00:00
parent 233a829bd7
commit af1be7f740
3 changed files with 7 additions and 4 deletions

View File

@@ -4,7 +4,6 @@
from copy import deepcopy
def get_combination(space, num):
combs = []
for i in range(num):
@@ -19,7 +18,6 @@ def get_combination(space, num):
new_combs.append( xstring )
combs = new_combs
return combs
class Structure:

View File

@@ -123,7 +123,7 @@ class NASBench201API(NASBenchMetaAPI):
"""
if self.verbose:
print('Call query_info_str_by_arch with arch={:} and hp={:}'.format(arch, hp))
self._query_info_str_by_arch(arch, hp, print_information)
return self._query_info_str_by_arch(arch, hp, print_information)
# obtain the metric for the `index`-th architecture
# `dataset` indicates the dataset: