aboutsummaryrefslogtreecommitdiff
path: root/nerv/examples/lmptb/lstmlm_ptb_main.lua
diff options
context:
space:
mode:
authortxh18 <cloudygooseg@gmail.com>2015-12-03 12:48:49 +0800
committertxh18 <cloudygooseg@gmail.com>2015-12-03 12:48:49 +0800
commitbba82cac04474b8177ab45d41543bc993801a4e0 (patch)
tree4ec102da65f39d778b1ec9631a123540453b2158 /nerv/examples/lmptb/lstmlm_ptb_main.lua
parent63cd5b0ab0d2fd1693fdaec0e57b5e02ad718dfb (diff)
moved tnn to main nerv dir and added it to Makefile
Diffstat (limited to 'nerv/examples/lmptb/lstmlm_ptb_main.lua')
-rw-r--r--nerv/examples/lmptb/lstmlm_ptb_main.lua6
1 files changed, 3 insertions, 3 deletions
diff --git a/nerv/examples/lmptb/lstmlm_ptb_main.lua b/nerv/examples/lmptb/lstmlm_ptb_main.lua
index 53a7bd5..4123378 100644
--- a/nerv/examples/lmptb/lstmlm_ptb_main.lua
+++ b/nerv/examples/lmptb/lstmlm_ptb_main.lua
@@ -2,7 +2,7 @@ require 'lmptb.lmvocab'
require 'lmptb.lmfeeder'
require 'lmptb.lmutil'
require 'lmptb.layer.init'
-require 'tnn.init'
+--require 'tnn.init'
require 'lmptb.lmseqreader'
require 'lm_trainer'
@@ -197,10 +197,10 @@ global_conf = {
hidden_size = 300, --set to 400 for a stable good test PPL
chunk_size = 15,
batch_size = 10,
- max_iter = 35,
+ max_iter = 45,
decay_iter = 10,
param_random = function() return (math.random() / 5 - 0.1) end,
- dropout_str = "0.5*15:0",
+ dropout_str = "0.5",
train_fn = train_fn,
valid_fn = valid_fn,