diff options
| -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 |