diff options
| author | txh18 <cloudygooseg@gmail.com> | 2015-11-25 23:42:37 +0800 |
|---|---|---|
| committer | txh18 <cloudygooseg@gmail.com> | 2015-11-25 23:42:37 +0800 |
| commit | 75a2d6a2a08caf987017f5a9043ac93afcd70980 (patch) | |
| tree | d3b08fa846de56f3cd8a3f9a7bd3add200840ea8 /nerv/examples/lmptb/tnn/tnn.lua | |
| parent | ca3500f01ea7ce695a4dbf70d2be8244827097c9 (diff) | |
changed auto-generating params, won not save in global_conf.param
Diffstat (limited to 'nerv/examples/lmptb/tnn/tnn.lua')
| -rw-r--r-- | nerv/examples/lmptb/tnn/tnn.lua | 4 |
1 files changed, 2 insertions, 2 deletions
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) |