diff options
Diffstat (limited to 'nerv/layer/rnn.lua')
-rw-r--r-- | nerv/layer/rnn.lua | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/nerv/layer/rnn.lua b/nerv/layer/rnn.lua index 38f2326..e59cf5b 100644 --- a/nerv/layer/rnn.lua +++ b/nerv/layer/rnn.lua @@ -1,10 +1,7 @@ local RNNLayer = nerv.class('nerv.RNNLayer', 'nerv.GraphLayer') function RNNLayer:__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 = layer_conf.gconf + nerv.Layer.__init(self, id, global_conf, layer_conf) self:check_dim_len(1, 1) local din = layer_conf.dim_in[1] @@ -12,7 +9,7 @@ function RNNLayer:__init(id, global_conf, layer_conf) local pr = layer_conf.pr if pr == nil then - pr = nerv.ParamRepo() + pr = nerv.ParamRepo({}, self.loc_type) end local layers = { |