update

scripts-rnn/train-WT2.sh

@@ -1,21 +1,14 @@
 #!/usr/bin/env sh
-if [ "$#" -ne 2 ] ;then
+if [ "$#" -ne 1 ] ;then
   echo "Input illegal number of parameters " $#
-  echo "Need 2 parameters for the GPU and the architecture"
+  echo "Need 1 parameters for the architectures"
   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
-arch=$2
-SAVED=./snapshots/NAS-RNN/Search-${arch}-WT2
+arch=$1
+SAVED=./output/NAS-RNN/Search-${arch}-WT2
 
-CUDA_VISIBLE_DEVICES=${gpus} python ./exps-nas/rnn/train_rnn_base.py \
+python ./exps-rnn/train_rnn_base.py \
 	--arch ${arch} \
 	--save_path ${SAVED} \
 	--config_path ./configs/NAS-WT2-BASE.config \