diff options
| author | txh18 <cloudygooseg@gmail.com> | 2015-10-25 22:33:04 +0800 |
|---|---|---|
| committer | txh18 <cloudygooseg@gmail.com> | 2015-10-25 22:33:04 +0800 |
| commit | 78feac4ca2c9a32710985dc0295c24d418712009 (patch) | |
| tree | d4aee9437868e2609865049852d1f9b639af88fe | |
| parent | 8e7d9453520840b6a5e269a101ca72b4a7ab36fa (diff) | |
added timer for profiling
| -rw-r--r-- | nerv/examples/lmptb/main.lua | 2 | ||||
| -rw-r--r-- | nerv/nn/layer_dag.lua | 2 |
2 files changed, 3 insertions, 1 deletions
diff --git a/nerv/examples/lmptb/main.lua b/nerv/examples/lmptb/main.lua index 74ce407..9b39e83 100644 --- a/nerv/examples/lmptb/main.lua +++ b/nerv/examples/lmptb/main.lua @@ -268,7 +268,7 @@ if (set == "ptb") then hidden_size = 200, batch_size = 10, - bptt = 6, --train bptt_block's words. could be set to zero + bptt = 3, --train bptt_block's words. could be set to zero max_iter = 18, param_random = function() return (math.random() / 5 - 0.1) end, independent = true, diff --git a/nerv/nn/layer_dag.lua b/nerv/nn/layer_dag.lua index 73bb77d..91818d6 100644 --- a/nerv/nn/layer_dag.lua +++ b/nerv/nn/layer_dag.lua @@ -262,7 +262,9 @@ function DAGLayer:propagate(input, output) for i = 1, #self.queue do local ref = self.queue[i] -- print(ref.layer.id) + self.gconf.timer:tic("(propagate)"..ref.layer.id); ret = ref.layer:propagate(ref.inputs, ref.outputs) + self.gconf.timer:toc("(propagate)"..ref.layer.id); end return ret end |