aboutsummaryrefslogtreecommitdiff
path: root/nerv/nn
diff options
context:
space:
mode:
authortxh18 <cloudygooseg@gmail.com>2015-10-27 16:24:55 +0800
committertxh18 <cloudygooseg@gmail.com>2015-10-27 16:24:55 +0800
commit7c95640c95f1cc1d84b4d49fa97fd922748b88a7 (patch)
treee78fc611b8768ddf6e9191d597bf667e83b3353b /nerv/nn
parentba8a1c9d5366c22b0b631f26ae1de7c5da2cbaeb (diff)
added update_select_rows for select_linear:update speed-up
Diffstat (limited to 'nerv/nn')
-rw-r--r--nerv/nn/layer_dag.lua2
1 files changed, 2 insertions, 0 deletions
diff --git a/nerv/nn/layer_dag.lua b/nerv/nn/layer_dag.lua
index 91818d6..4904f4f 100644
--- a/nerv/nn/layer_dag.lua
+++ b/nerv/nn/layer_dag.lua
@@ -251,7 +251,9 @@ function DAGLayer:update(bp_err, input, output)
-- print("update")
for id, ref in pairs(self.queue) do
-- print(ref.layer.id)
+ self.gconf.timer:tic("(update)"..ref.layer.id);
ref.layer:update(ref.err_inputs, ref.inputs, ref.outputs)
+ self.gconf.timer:toc("(update)"..ref.layer.id);
end
end