iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update README

Browse Source
This commit is contained in:
D-X-Y
2019-12-28 15:42:36 +11:00
parent d791622b63
commit 4c144b7437
6 changed files with 59 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/algos/R_EA.py
View File

@@ -59,7 +59,7 @@ def train_and_eval(arch, nas_bench, extra_info):
if nas_bench is not None:
arch_index = nas_bench.query_index_by_arch( arch )
assert arch_index >= 0, 'can not find this arch : {:}'.format(arch)
info = nas_bench.get_more_info(arch_index, 'cifar10-valid', True)
info = nas_bench.get_more_info(arch_index, 'cifar10-valid', None, True)
valid_acc, time_cost = info['valid-accuracy'], info['train-all-time'] + info['valid-per-time']
#_, valid_acc = info.get_metrics('cifar10-valid', 'x-valid' , 25, True) # use the validation accuracy after 25 training epochs
else: