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 \