aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--nerv/examples/lmptb/bilstmlm_ptb_main.lua5
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"]