diff options
| -rw-r--r-- | nerv/examples/lmptb/grulm_ptb_main.lua | 14 |
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 |