From 75a2d6a2a08caf987017f5a9043ac93afcd70980 Mon Sep 17 00:00:00 2001
From: txh18 <cloudygooseg@gmail.com>
Date: Wed, 25 Nov 2015 23:42:37 +0800
Subject: changed auto-generating params, won not save in global_conf.param

---
 nerv/examples/lmptb/tnn/tnn.lua | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'nerv/examples/lmptb/tnn/tnn.lua')

diff --git a/nerv/examples/lmptb/tnn/tnn.lua b/nerv/examples/lmptb/tnn/tnn.lua
index c2e397c..c87f963 100644
--- a/nerv/examples/lmptb/tnn/tnn.lua
+++ b/nerv/examples/lmptb/tnn/tnn.lua
@@ -1,4 +1,4 @@
-local TNN = nerv.class("nerv.TNN", "nerv.Layer")
+local TNN = nerv.class("nerv.TNN")
 
 local function parse_id(str)
     --used to parse layerid[portid],time
@@ -541,7 +541,7 @@ end
 --Return: nerv.ParamRepo
 function TNN:get_params()
     local param_repos = {}
-    for id, ref in pairs(self.queue) do
+    for id, ref in pairs(self.layers) do
         table.insert(param_repos, ref.layer:get_params())
     end
     return nerv.ParamRepo.merge(param_repos)
-- 
cgit v1.2.3