aboutsummaryrefslogtreecommitdiff
path: root/nerv/examples/lmptb/lm_trainer.lua
diff options
context:
space:
mode:
authortxh18 <cloudygooseg@gmail.com>2015-11-15 22:44:02 +0800
committertxh18 <cloudygooseg@gmail.com>2015-11-15 22:44:02 +0800
commitef40688d5a0a3b7eae18dc364a40ae4e8e7619e7 (patch)
treeec12c403ddc5a2c0d0928f3a8249e74b4c7d5916 /nerv/examples/lmptb/lm_trainer.lua
parent5760914d95059777c5e475f3c42d1b32983235a3 (diff)
added timer
Diffstat (limited to 'nerv/examples/lmptb/lm_trainer.lua')
-rw-r--r--nerv/examples/lmptb/lm_trainer.lua4
1 files changed, 4 insertions, 0 deletions
diff --git a/nerv/examples/lmptb/lm_trainer.lua b/nerv/examples/lmptb/lm_trainer.lua
index d34634c..226873b 100644
--- a/nerv/examples/lmptb/lm_trainer.lua
+++ b/nerv/examples/lmptb/lm_trainer.lua
@@ -63,6 +63,10 @@ function LMTrainer.lm_process_file(global_conf, fn, tnn, do_train)
next_log_wcn = next_log_wcn + global_conf.log_w_num
printf("%s %d words processed %s.\n", global_conf.sche_log_pre, result["rnn"].cn_w, os.date())
printf("\t%s log prob per sample :%f.\n", global_conf.sche_log_pre, result:logp_sample("rnn"))
+ for key, value in pairs(global_conf.timer.rec) do
+ printf("\t [global_conf.timer]: time spent on %s:%.5fs\n", key, value)
+ end
+ global_conf.timer:flush()
nerv.LMUtil.wait(0.1)
end