diff options
Diffstat (limited to 'nerv/layer/identity.lua')
| -rw-r--r-- | nerv/layer/identity.lua | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/nerv/layer/identity.lua b/nerv/layer/identity.lua index aeeff89..d56337d 100644 --- a/nerv/layer/identity.lua +++ b/nerv/layer/identity.lua @@ -1,10 +1,7 @@ local IdentityLayer = nerv.class('nerv.IdentityLayer', 'nerv.Layer') function IdentityLayer:__init(id, global_conf, layer_conf) - self.id = id - self.dim_in = layer_conf.dim_in - self.dim_out = layer_conf.dim_out - self.gconf = global_conf + nerv.Layer.__init(self, id, global_conf, layer_conf) self:check_dim_len(1, 1) if self.dim_in[1] ~= self.dim_out[1] then nerv.error('mismatching dimensions of input and output') @@ -29,5 +26,5 @@ function IdentityLayer:update() end function IdentityLayer:get_params() - return nerv.ParamRepo({}) + return nerv.ParamRepo({}, self.loc_type) end |