first commit

NAS-Bench-201/script/download_preprocessed_dataset.sh

@@ -0,0 +1,4 @@
+export LD_LIBRARY_PATH=/opt/conda/envs/gtctnz_2/lib/python3.7/site-packages/nvidia/cublas/lib/
+
+echo '[Downloading processed]'
+python main_exp/transfer_nag/get_files/get_preprocessed_data.py

NAS-Bench-201/script/download_raw_dataset.sh

@@ -0,0 +1,15 @@
+export LD_LIBRARY_PATH=/opt/conda/envs/gtctnz_2/lib/python3.7/site-packages/nvidia/cublas/lib/
+
+DATANAME=$1
+
+if [[ $DATANAME = 'aircraft' ]]; then
+    echo '[Downloading aircraft]'
+    python main_exp/transfer_nag/get_files/get_aircraft.py
+
+elif [[ $DATANAME = 'pets' ]]; then
+    echo '[Downloading pets]'
+    python main_exp/transfer_nag/get_files/get_pets.py
+
+else
+    echo 'Not Implemeted'
+fi

NAS-Bench-201/script/tr_meta_surrogate.sh

@@ -0,0 +1,6 @@
+FOLDER_NAME='tr_meta_surrogate_nb201'
+
+CUDA_VISIBLE_DEVICES=$1 python main.py --config configs/tr_meta_surrogate.py \
+    --mode train \
+    --config.folder_name $FOLDER_NAME
+

NAS-Bench-201/script/tr_scorenet.sh

@@ -0,0 +1,5 @@
+FOLDER_NAME='tr_scorenet_nb201'
+
+CUDA_VISIBLE_DEVICES=$1 python main.py --config configs/tr_scorenet.py \
+    --mode train \
+    --config.folder_name $FOLDER_NAME

NAS-Bench-201/script/transfer_nag.sh

@@ -0,0 +1,10 @@
+FOLDER_NAME='transfer_nag_nb201'
+
+GPU=$1
+DATANAME=$2
+
+CUDA_VISIBLE_DEVICES=$GPU python main_exp/transfer_nag/main.py \
+    --gpu $GPU \
+    --test \
+    --folder_name $FOLDER_NAME \
+    --data-name $DATANAME