summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authortxh18 <[email protected]>2016-01-01 11:51:23 +0800
committertxh18 <[email protected]>2016-01-01 11:51:23 +0800
commit464cd37d218ac7dee90d44e721655a7c79ebe961 (patch)
treefc75462f69a5022b40a182126de72a4fc116b752
parentc0e533cef55138e0de201ca9da9c0e3f3b8a61f1 (diff)
...
-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'