diff options
-rw-r--r-- | nerv/examples/lmptb/grulm_ptb_main.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/nerv/examples/lmptb/grulm_ptb_main.lua b/nerv/examples/lmptb/grulm_ptb_main.lua index b2b3cba..fc36a92 100644 --- a/nerv/examples/lmptb/grulm_ptb_main.lua +++ b/nerv/examples/lmptb/grulm_ptb_main.lua @@ -183,9 +183,10 @@ local train_fn, valid_fn, test_fn global_conf = {} local set = arg[1] --"test" +root_dir = '/home/slhome/txh18/workspace' + if (set == "ptb") then -root_dir = '/home/slhome/txh18/workspace' data_dir = root_dir .. '/ptb/DATA' train_fn = data_dir .. '/ptb.train.txt.adds' valid_fn = data_dir .. '/ptb.valid.txt.adds' @@ -257,7 +258,6 @@ global_conf = { elseif (set == "twitter") then -root_dir = '/home/slhome/txh18/workspace' data_dir = root_dir .. '/twitter_new/DATA' train_fn = data_dir .. '/twitter.choose2.adds' valid_fn = data_dir .. '/twitter.valid.adds' |