This commit is contained in:
dongxuanyi
2019-03-28 23:21:03 +08:00
parent d6e9078568
commit a94a67b55d
2 changed files with 3 additions and 2 deletions

View File

@@ -15,7 +15,7 @@ gpus=$1
arch=$2
SAVED=./snapshots/NAS-RNN/Search-${arch}-PTB
CUDA_VISIBLE_DEVICES=${gpus} python ./exps-nas/rnn/train_rnn_base.py \
CUDA_VISIBLE_DEVICES=${gpus} python ./exps-rnn/train_rnn_base.py \
--arch ${arch} \
--save_path ${SAVED} \
--config_path ./configs/NAS-PTB-BASE.config \