diff options
Diffstat (limited to 'nerv/layer/affine.lua')
| -rw-r--r-- | nerv/layer/affine.lua | 4 |
1 files changed, 2 insertions, 2 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 |