Update VIS-CODES and SCRIPTS

This commit is contained in:
D-X-Y 2020-07-22 21:59:47 +00:00
parent a2a1abcb7d
commit cb471a8966
2 changed files with 6 additions and 5 deletions

View File

@ -32,10 +32,11 @@ if [ "$alg_type" == "mul" ]; then
python exps/experimental/vis-bench-algos.py --search_space sss
else
seeds="777 888 999"
algos="darts-v1 darts-v2 gdas setn random enas"
epoch=200
for seed in ${seeds}
do
for alg in "darts-v1 darts-v2 gdas setn random enas"
for alg in ${algos}
do
python ./exps/algos-v2/search-cell.py --dataset cifar10 --data_path $TORCH_HOME/cifar.python --algo ${alg} --rand_seed ${seed} --overwite_epochs ${epoch}
python ./exps/algos-v2/search-cell.py --dataset cifar100 --data_path $TORCH_HOME/cifar.python --algo ${alg} --rand_seed ${seed} --overwite_epochs ${epoch}

View File

@ -32,8 +32,8 @@ def fetch_data(root_dir='./output/search', search_space='tss', dataset=None):
seeds = [777]
alg2name['GDAS'] = 'gdas-affine0_BN0-None'
alg2name['RSPS'] = 'random-affine0_BN0-None'
alg2name['DARTS (1st)'] = 'darts-v1-affine0_BN0-None'
"""
alg2name['DARTS (1st)'] = 'darts-v1-affine1_BN0-None'
alg2name['DARTS (2nd)'] = 'darts-v2-affine1_BN0-None'
alg2name['SETN'] = 'setn-affine1_BN0-None'
"""