lstm_tnn can be run, todo:testing

1 files changed, 12 insertions, 5 deletions

diff --git a/nerv/examples/lmptb/lstmlm_ptb_main.lua b/nerv/examples/lmptb/lstmlm_ptb_main.lua
index d3f38a2..42b541f 100644
--- a/nerv/examples/lmptb/lstmlm_ptb_main.lua
+++ b/nerv/examples/lmptb/lstmlm_ptb_main.lua
@@ -129,10 +129,17 @@ function prepare_tnn(global_conf, layerRepo)
     --input: input_w, input_w, ... input_w_now, last_activation
     local connections_t = {
         {"<input>[1]", "selectL1[1]", 0},
-        {"selectL1[1]", "recurrentL1[1]", 0},  
-        {"recurrentL1[1]", "sigmoidL1[1]", 0},
-        {"sigmoidL1[1]", "combinerL1[1]", 0},
-        {"combinerL1[1]", "recurrentL1[2]", 1},
+        
+        --{"selectL1[1]", "recurrentL1[1]", 0},  
+        --{"recurrentL1[1]", "sigmoidL1[1]", 0},
+        --{"sigmoidL1[1]", "combinerL1[1]", 0},
+        --{"combinerL1[1]", "recurrentL1[2]", 1},
+        
+        {"selectL1[1]", "lstmL1[1]", 0},
+        {"lstmL1[2]", "lstmL1[3]", 1},
+        {"lstmL1[1]", "combinerL1[1]", 0},
+        {"combinerL1[1]", "lstmL1[2]", 1},
+
         {"combinerL1[2]", "outputL[1]", 0},
         {"outputL[1]", "softmaxL[1]", 0},
         {"<input>[2]", "softmaxL[2]", 0},
@@ -268,7 +275,7 @@ if (arg[2] ~= nil) then
     loadstring(arg[2])() 
     nerv.LMUtil.wait(0.5)
 else
-    printf("%s not user setting, all default...\n", global_conf.sche_log_pre)
+    printf("%s no user setting, all default...\n", global_conf.sche_log_pre)
 end
 
 global_conf.work_dir = global_conf.work_dir_base .. 'h' .. global_conf.hidden_size .. 'ch' .. global_conf.chunk_size .. 'ba' .. global_conf.batch_size .. 'slr' ..   global_conf.lrate .. 'wc' .. global_conf.wcost