update scripts

This commit is contained in:
Xuanyi Dong
2019-03-30 02:10:20 +08:00
parent 3734384b68
commit c8dddf9cf9
9 changed files with 61 additions and 23 deletions

View File

@@ -1,4 +1,4 @@
#!/usr/bin/env sh
#!/bin/bash
if [ "$#" -ne 1 ] ;then
echo "Input illegal number of parameters " $#
echo "Need 1 parameters for the architectures"
@@ -7,8 +7,18 @@ fi
arch=$1
SAVED=./output/NAS-RNN/Search-${arch}-WT2
PY_C="./env/bin/python"
python ./exps-rnn/train_rnn_base.py \
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-WT2-BASE.config \