diff options
Diffstat (limited to 'nerv/layer/init.lua')
| -rw-r--r-- | nerv/layer/init.lua | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/nerv/layer/init.lua b/nerv/layer/init.lua index c5b7657..3a6cbcd 100644 --- a/nerv/layer/init.lua +++ b/nerv/layer/init.lua @@ -114,12 +114,13 @@ function Layer:find_param(plist, lconf, gconf, p_type, p_dim, p_gen) if lconf.pr:has_param(pid) then return lconf.pr:get_param(pid) end + pid = self.id .. '_' .. pname + if lconf.pr:has_param(pid) then + nerv.info("param id for [%s] of layer [%s] is generated automatically.", plist[1], self.id) + return lconf.pr:get_param(pid) + end end pid = self.id .. '_' .. plist[1] - if lconf.pr:has_param(pid) then - nerv.info("param id for [%s] of layer [%s] is generated automatically.", plist[1], self.id) - return lconf.pr:get_param(pid) - end nerv.info("param id for [%s] of layer [%s] is not found in the specified param repo, " .. "switch to auto-generate", plist_str, self.id) local p = p_type(pid, gconf) |