Update docs of NATS-Bench
This commit is contained in:
@@ -33,7 +33,7 @@ def main(xargs, api):
|
||||
logger.log('{:} use api : {:}'.format(time_string(), api))
|
||||
api.reset_time()
|
||||
|
||||
search_space = get_search_spaces(xargs.search_space, 'nas-bench-301')
|
||||
search_space = get_search_spaces(xargs.search_space, 'nats-bench')
|
||||
if xargs.search_space == 'tss':
|
||||
random_arch = random_topology_func(search_space)
|
||||
else:
|
||||
|
Reference in New Issue
Block a user