From 5e34a22d44fcd74df7378aca0de050cdd349c8b9 Mon Sep 17 00:00:00 2001 From: txh18 Date: Tue, 29 Dec 2015 18:30:39 +0800 Subject: small bug fix in tnn --- nerv/tnn/tnn.lua | 4 ++-- 1 file 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 -- cgit v1.2.3