From a94a67b55d2816d20304565948bc4c74f480d432 Mon Sep 17 00:00:00 2001 From: dongxuanyi Date: Thu, 28 Mar 2019 23:21:03 +0800 Subject: [PATCH] update --- exps-rnn/train_rnn_base.py | 3 ++- scripts-rnn/train-PTB.sh | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/exps-rnn/train_rnn_base.py b/exps-rnn/train_rnn_base.py index 7407541..518b78f 100644 --- a/exps-rnn/train_rnn_base.py +++ b/exps-rnn/train_rnn_base.py @@ -8,7 +8,8 @@ import torchvision.datasets as dset import torch.backends.cudnn as cudnn import torchvision.transforms as transforms from pathlib import Path -lib_dir = (Path(__file__).parent / '..' / '..' / 'lib').resolve() +lib_dir = (Path(__file__).parent / '..' / 'lib').resolve() +print ('lib-dir : {:}'.format(lib_dir)) if str(lib_dir) not in sys.path: sys.path.insert(0, str(lib_dir)) from utils import AverageMeter, time_string, time_file_str, convert_secs2time from utils import print_log, obtain_accuracy diff --git a/scripts-rnn/train-PTB.sh b/scripts-rnn/train-PTB.sh index bd0b5cd..661dced 100644 --- a/scripts-rnn/train-PTB.sh +++ b/scripts-rnn/train-PTB.sh @@ -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 \