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 /nerv/nn | |
| parent | 8e7d9453520840b6a5e269a101ca72b4a7ab36fa (diff) | |
added timer for profiling
Diffstat (limited to 'nerv/nn')
| -rw-r--r-- | nerv/nn/layer_dag.lua | 2 |
1 files changed, 2 insertions, 0 deletions
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 |