update NAS-Bench-102 baselines / support track_running_stats

2019-12-23 13:32:20 +11:00
parent 729ce136db
commit 2dc8dce6d3
9 changed files with 56 additions and 57 deletions
--- a/scripts-search/NAS-Bench-102/train-models.sh
+++ b/scripts-search/NAS-Bench-102/train-models.sh
@@ -28,7 +28,7 @@ else
  mode=cover
 fi

-OMP_NUM_THREADS=4 python ./exps/AA-NAS-Bench-main.py \
+OMP_NUM_THREADS=4 python ./exps/NAS-Bench-102/main.py \
 	--mode ${mode} --save_dir ${save_dir} --max_node 4 \
 	--use_less ${use_less} \
 	--datasets cifar10 cifar10 cifar100 ImageNet16-120 \