autodl-projects/scripts-rnn/train-PTB.sh

#!/usr/bin/env sh
if [ "$#" -ne 1 ] ;then
  echo "Input illegal number of parameters " $#
  echo "Need 1 parameters for the GPU and the architecture"
  exit 1               
fi 

arch=$1
SAVED=./output/NAS-RNN/Search-${arch}-PTB
PY_C="./env/bin/python"

if [ ! -f ${PY_C} ]; then
  echo "Local Run with Python: "`which python`
  PY_C="python"
else
  echo "Cluster Run with Python: "${PY_C}
fi

${PY_C} --version

${PY_C} ./exps-rnn/train_rnn_base.py \
	--arch ${arch} \
	--save_path ${SAVED} \
	--config_path ./configs/NAS-PTB-BASE.config \
	--print_freq 200