3 files changed, 25 insertions, 4 deletions
diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua
index 6903c51..e24a0c6 100644
--- a/nerv/layer/affine.lua
+++ b/nerv/layer/affine.lua
@@ -61,10 +61,10 @@ end
 
 function AffineLayer:__init(id, global_conf, layer_conf)
     self.id = id
-    self.ltp = layer_conf.ltp
-    self.bp = layer_conf.bp
     self.dim_in = layer_conf.dim_in
     self.dim_out = layer_conf.dim_out
+    self.ltp = self:find_param("ltp", layer_conf, global_conf, nerv.LinearTransParam, {self.dim_in[1], self.dim_out[1]}) --layer_conf.ltp
+    self.bp = self:find_param("bp", layer_conf, global_conf, nerv.BiasParam, {1, self.dim_out[1]})--layer_conf.bp
     self.gconf = global_conf
     self:check_dim_len(1, 1) -- exactly one input and one output
     -- self.direct_update = layer_conf.direct_update or global_conf.direct_update
diff --git a/nerv/layer/affine_recurrent.lua b/nerv/layer/affine_recurrent.lua
index da189e0..d537f4a 100644
--- a/nerv/layer/affine_recurrent.lua
+++ b/nerv/layer/affine_recurrent.lua
@@ -10,8 +10,8 @@ function Recurrent:__init(id, global_conf, layer_conf)
     self.dim_out = layer_conf.dim_out
     self.gconf = global_conf
 
-    self.bp = layer_conf.bp
-    self.ltp_hh = layer_conf.ltp_hh --from hidden to hidden
+    self.bp = self:find_param("bp", layer_conf, global_conf, nerv.BiasParam, {1, self.dim_out[1]}) --layer_conf.bp
+    self.ltp_hh = self:find_param("ltp_hh", layer_conf, global_conf, nerv.LinearTransParam, {self.dim_in[2], self.dim_out[1]}) --layer_conf.ltp_hh --from hidden to hidden
     
     self:check_dim_len(2, 1)
     self.direct_update = layer_conf.direct_update
diff --git a/nerv/layer/init.lua b/nerv/layer/init.lua
index 6861b0e..c6d0a98 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
+    if pr:has_param(pid_g) == true then
+        nerv.printf("Param [%s] of layer [%s] found in paramRepo.\n", pid, self.id)
+        p = pr:get_param(pid_g)
+        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(gconf.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')