init
This commit is contained in:
		| @@ -1,12 +0,0 @@ | ||||
| # Search RNN cell | ||||
| ``` | ||||
| bash scripts-nas-rnn/search-baseline.sh 3 | ||||
| bash scripts-nas-rnn/search-accelerate.sh 0 200 10 1 | ||||
| ``` | ||||
|  | ||||
| # Train the Searched Model | ||||
| ``` | ||||
| bash scripts-nas-rnn/train-PTB.sh 3 DARTS_V1 | ||||
| bash scripts-nas-rnn/train-WT2.sh 3 DARTS_V1 | ||||
| bash scripts-nas-rnn/train-PTB.sh 3 DARTS_V2 | ||||
| ``` | ||||
| @@ -1,26 +0,0 @@ | ||||
| #!/usr/bin/env sh | ||||
| if [ "$#" -ne 4 ] ;then | ||||
|   echo "Input illegal number of parameters " $# | ||||
|   echo "Need 4 parameters for the GPU and the epochs and tau-max and tau-min" | ||||
|   exit 1                | ||||
| fi  | ||||
| if [ "$TORCH_HOME" = "" ]; then | ||||
|   echo "Must set TORCH_HOME envoriment variable for data dir saving" | ||||
|   exit 1 | ||||
| else | ||||
|   echo "TORCH_HOME : $TORCH_HOME" | ||||
| fi | ||||
|  | ||||
| gpus=$1 | ||||
| epoch=$2 | ||||
| tau_max=$3 | ||||
| tau_min=$4 | ||||
| SAVED=./snapshots/NAS-RNN/Search-Accelerate-tau_${tau_max}_${tau_min}-${epoch} | ||||
|  | ||||
| CUDA_VISIBLE_DEVICES=${gpus} python ./exps-nas/rnn/acc_rnn_search.py \ | ||||
| 	--data_path ./data/data/penn \ | ||||
| 	--save_path ${SAVED} \ | ||||
| 	--epochs ${epoch} \ | ||||
| 	--tau_max ${tau_max} --tau_min ${tau_min} \ | ||||
| 	--config_path ./configs/NAS-PTB-BASE.config \ | ||||
| 	--print_freq 200 | ||||
| @@ -1,23 +0,0 @@ | ||||
| #!/usr/bin/env sh | ||||
| if [ "$#" -ne 1 ] ;then | ||||
|   echo "Input illegal number of parameters " $# | ||||
|   echo "Need 1 parameters for the GPU" | ||||
|   exit 1                | ||||
| fi  | ||||
| if [ "$TORCH_HOME" = "" ]; then | ||||
|   echo "Must set TORCH_HOME envoriment variable for data dir saving" | ||||
|   exit 1 | ||||
| else | ||||
|   echo "TORCH_HOME : $TORCH_HOME" | ||||
| fi | ||||
|  | ||||
| gpus=$1 | ||||
| epoch=50 | ||||
| SAVED=./snapshots/NAS-RNN/Search-Baseline-${epoch} | ||||
|  | ||||
| CUDA_VISIBLE_DEVICES=${gpus} python ./exps-nas/rnn/train_rnn_search.py \ | ||||
| 	--data_path ./data/data/penn \ | ||||
| 	--save_path ${SAVED} \ | ||||
| 	--epochs ${epoch} \ | ||||
| 	--config_path ./configs/NAS-PTB-BASE.config \ | ||||
| 	--print_freq 200 | ||||
		Reference in New Issue
	
	Block a user