aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--nerv/examples/lmptb/grulm_ptb_main.lua4
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'