working on automatic parameter for layers

1 files changed, 21 insertions, 0 deletions

diff --git a/nerv/layer/init.lua b/nerv/layer/init.lua
index 6861b0e..d268caa 100644
--- a/nerv/layer/init.lua
+++ b/nerv/layer/init.lua
@@ -70,6 +70,27 @@ function Layer:get_dim()
     return self.dim_in, self.dim_out
 end
 
+function Layer:find_param(pid, l_conf, gconf, p_type, p_dim)
+    if l_conf[pid] ~= nil then
+        nerv.printf("Param [%s] of layer [%s] found in layer_conf.\n", pid, self.id)
+        return l_conf[pid]
+    end
+    local pid_g = self.id .. '_' .. pid --global identifier
+    local pr = gconf.paramRepo
+    local p
+    p = pr:get_param(pid_g)
+    if p ~= nil then
+        nerv.printf("Param [%s] of layer [%s] found in paramRepo.\n", pid, self.id)
+        return p
+    end
+    nerv.printf("Param [%s] of layer [%s] is not found in layer_conf or paramRepo, switch to auto-generate.\n", pid, self.id)
+    p = p_type(pid_g, gconf)
+    p.trans = gconf.cumat_type(unpack(p_dim))
+    p.trans:generate(global_conf.param_random)
+    pr:add(pid_g, p) --add the parameter into the paramRepo
+    return p
+end
+
 nerv.include('affine.lua')
 nerv.include('sigmoid.lua')
 nerv.include('softmax_ce.lua')