diff options
| author | txh18 <[email protected]> | 2015-12-29 18:30:39 +0800 |
|---|---|---|
| committer | txh18 <[email protected]> | 2015-12-29 18:30:39 +0800 |
| commit | 5e34a22d44fcd74df7378aca0de050cdd349c8b9 (patch) | |
| tree | 8c712ee2241b3f949c07e0b87efea82a4b99e66f | |
| parent | e68d606fbea209794d7380e84dbffc1c6d22021e (diff) | |
small bug fix in tnn
| -rw-r--r-- | nerv/tnn/tnn.lua | 4 |
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 |