diff options
| author | txh18 <cloudygooseg@gmail.com> | 2015-11-20 19:58:14 +0800 |
|---|---|---|
| committer | txh18 <cloudygooseg@gmail.com> | 2015-11-20 19:58:14 +0800 |
| commit | 4f5b45b79b8d5f6a9094888cf6b929fe86ac24a3 (patch) | |
| tree | 60d5b6232b8d33da9178717c27ecb6dc3591d0b6 /nerv/layer/affine.lua | |
| parent | 6456f5d0b97c5ad7e35c58751f74b8c9fefb635e (diff) | |
working on automatic parameter for layers
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 6a541e8..3e84ec0 100644 --- a/nerv/layer/affine.lua +++ b/nerv/layer/affine.lua @@ -63,10 +63,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 |