diff options
| author | txh18 <[email protected]> | 2015-12-08 11:14:54 +0800 |
|---|---|---|
| committer | txh18 <[email protected]> | 2015-12-08 11:14:54 +0800 |
| commit | aa18efcb0b809993274bc4560b64017111dfd71c (patch) | |
| tree | a00b60bcacbf112250c5568d3c7ff34448c82e4a | |
| parent | 775d35373980b357ae77c9694f71d98e8ef103d6 (diff) | |
small bugs fix for bilstm script
| -rw-r--r-- | nerv/examples/lmptb/bilstmlm_ptb_main.lua | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/nerv/examples/lmptb/bilstmlm_ptb_main.lua b/nerv/examples/lmptb/bilstmlm_ptb_main.lua index b896bed..7e238b3 100644 --- a/nerv/examples/lmptb/bilstmlm_ptb_main.lua +++ b/nerv/examples/lmptb/bilstmlm_ptb_main.lua @@ -500,7 +500,7 @@ end --if commands["test"] if commands["testout"] == 1 then nerv.printf("===TEST OUT===\n") nerv.printf("q_file:\t%s\n", q_file) - local q_fn = qdata_dir .. q_file + local q_fn = qdata_dir .. '/' .. q_file global_conf.sche_log_pre = "[SCHEDULER FINAL_TEST]:" if final_iter ~= -1 and test_iter == -1 then test_iter = final_iter @@ -510,7 +510,8 @@ if commands["testout"] == 1 then end tnn = load_net(global_conf, test_iter) global_conf.dropout_rate = 0 - LMTrainer.lm_process_file_birnn(global_conf, q_fn, tnn, false) --false update! + LMTrainer.lm_process_file_birnn(global_conf, q_fn, tnn, false, + {["one_sen_report"] = true}) --false update! end --if commands["testout"] |