aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--nerv/examples/lmptb/grulm_ptb_main.lua14
1 files changed, 7 insertions, 7 deletions
diff --git a/nerv/examples/lmptb/grulm_ptb_main.lua b/nerv/examples/lmptb/grulm_ptb_main.lua
index c5c011a..831146b 100644
--- a/nerv/examples/lmptb/grulm_ptb_main.lua
+++ b/nerv/examples/lmptb/grulm_ptb_main.lua
@@ -259,10 +259,10 @@ elseif (set == "twitter") then
root_dir = '/home/slhome/txh18/workspace'
data_dir = root_dir .. '/twitter_new/DATA'
-train_fn = data_dir .. '/twitter.choose.adds'
+train_fn = data_dir .. '/twitter.choose2.adds'
valid_fn = data_dir .. '/twitter.valid.adds'
test_fn = data_dir .. '/comm.test.choose-ppl.adds'
-vocab_fn = data_dir .. '/twitter.choose.train.vocab'
+vocab_fn = data_dir .. '/vocab.25000'
--qdata_dir = root_dir .. '/ptb/questionGen/gen'
@@ -275,22 +275,22 @@ global_conf = {
hidden_size = 300,
layer_num = 1,
chunk_size = 15,
- batch_size = 20,
- max_iter = 35,
+ batch_size = 32,
+ max_iter = 30,
lr_decay = 1.003,
decay_iter = 10,
param_random = function() return (math.random() / 5 - 0.1) end,
- dropout_str = "0",
+ dropout_str = "0.5",
train_fn = train_fn,
valid_fn = valid_fn,
test_fn = test_fn,
vocab_fn = vocab_fn,
- max_sen_len = 90,
+ max_sen_len = 32,
sche_log_pre = "[SCHEDULER]:",
log_w_num = 40000, --give a message when log_w_num words have been processed
timer = nerv.Timer(),
- work_dir_base = root_dir .. '/twitter_new/EXP-nerv/lstmlm_v1.0'
+ work_dir_base = root_dir .. '/twitter_new/EXP-nerv/grulm_v1.0'
}
else