aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authortxh18 <[email protected]>2015-12-29 18:30:39 +0800
committertxh18 <[email protected]>2015-12-29 18:30:39 +0800
commit5e34a22d44fcd74df7378aca0de050cdd349c8b9 (patch)
tree8c712ee2241b3f949c07e0b87efea82a4b99e66f
parente68d606fbea209794d7380e84dbffc1c6d22021e (diff)
small bug fix in tnn
-rw-r--r--nerv/tnn/tnn.lua4
1 files changed, 2 insertions, 2 deletions
diff --git a/nerv/tnn/tnn.lua b/nerv/tnn/tnn.lua
index 95e23a9..e7d4981 100644
--- a/nerv/tnn/tnn.lua
+++ b/nerv/tnn/tnn.lua
@@ -113,8 +113,8 @@ function TNN:__init(id, global_conf, layer_conf)
local parsed_conns = {}
local _
- for id, layer in pairs(layer_conf.sub_layers.layers) do
- discover(id, layer, layer_conf.sub_layers)
+ for id, _ in pairs(layer_conf.sub_layers.layers) do --caution: with this line, some layer not connected will be included
+ discover(id, layers, layer_conf.sub_layers)
end
for _, ll in pairs(layer_conf.connections) do