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

update TEMPLATE of GITHUB

Browse Source
This commit is contained in:
D-X-Y
2019-12-23 17:06:14 +11:00
parent f7a38d9170
commit 34ff796e1a
86 changed files with 42 additions and 4672 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
exps/algos/GDAS.py
View File

@@ -104,7 +104,8 @@ def main(xargs):
search_space = get_search_spaces('cell', xargs.search_space_name)
model_config = dict2config({'name': 'GDAS', 'C': xargs.channel, 'N': xargs.num_cells,
'max_nodes': xargs.max_nodes, 'num_classes': class_num,
'space' : search_space}, None)
'space' : search_space,
'affine' : False, 'track_running_stats': True}, None)
search_model = get_cell_based_tiny_net(model_config)
logger.log('search-model :\n{:}'.format(search_model))
@@ -194,7 +195,7 @@ def main(xargs):
logger.log('\n' + '-'*100)
# check the performance from the architecture dataset
logger.log('DARTS-V1 : run {:} epochs, cost {:.1f} s, last-geno is {:}.'.format(total_epoch, search_time.sum, genotypes[total_epoch-1]))
logger.log('GDAS : run {:} epochs, cost {:.1f} s, last-geno is {:}.'.format(total_epoch, search_time.sum, genotypes[total_epoch-1]))
if api is not None: logger.log('{:}'.format( api.query_by_arch(genotypes[total_epoch-1]) ))
logger.close()