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

update NAS-Bench-102 baselines

Browse Source
This commit is contained in:
D-X-Y
2019-12-25 10:30:50 +11:00
parent 44a0d51449
commit 1d5e8debad
5 changed files with 48 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts-search/algos/BOHB.sh
View File

@@ -34,5 +34,6 @@ OMP_NUM_THREADS=4 python ./exps/algos/BOHB.py \
--dataset ${dataset} --data_path ${data_path} \
--search_space_name ${space} \
--arch_nas_dataset ${TORCH_HOME}/NAS-Bench-102-v1_0-e61699.pth \
--n_iters 6 --num_samples 3 \
--time_budget 12000 \
--n_iters 100 --num_samples 4 --random_fraction 0 \
--workers 4 --print_freq 200 --rand_seed ${seed}

3
scripts-search/algos/REINFORCE.sh
View File

@@ -34,5 +34,6 @@ OMP_NUM_THREADS=4 python ./exps/algos/reinforce.py \
--dataset ${dataset} --data_path ${data_path} \
--search_space_name ${space} \
--arch_nas_dataset ${TORCH_HOME}/NAS-Bench-102-v1_0-e61699.pth \
--learning_rate 0.001 --RL_steps 100 --EMA_momentum 0.9 \
--time_budget 12000 \
--learning_rate 0.001 --EMA_momentum 0.9 \
--workers 4 --print_freq 200 --rand_seed ${seed}