update NAS-Bench-102 baselines

2019-12-25 10:30:50 +11:00
parent 44a0d51449
commit 1d5e8debad
5 changed files with 48 additions and 17 deletions
--- a/scripts-search/algos/REINFORCE.sh
+++ b/scripts-search/algos/REINFORCE.sh
@@ -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}