diff options
| -rw-r--r-- | nerv/layer/affine.lua | 2 | ||||
| -rw-r--r-- | nerv/layer/lstm_gate.lua | 2 | ||||
| -rw-r--r-- | nerv/layer/projection.lua | 2 |
3 files changed, 3 insertions, 3 deletions
diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua index ace4a78..16250fd 100644 --- a/nerv/layer/affine.lua +++ b/nerv/layer/affine.lua @@ -172,7 +172,7 @@ end function AffineLayer:get_params() local pr = nerv.ParamRepo({self.ltp1, self.bp}, self.loc_type) for i = 2, #self.dim_in do - pr:add(self["ltp" .. i].id, self["ltp" .. i]) + pr:add(self["ltp" .. i]) end return pr end diff --git a/nerv/layer/lstm_gate.lua b/nerv/layer/lstm_gate.lua index 82824fa..39a3ff7 100644 --- a/nerv/layer/lstm_gate.lua +++ b/nerv/layer/lstm_gate.lua @@ -91,7 +91,7 @@ end function LSTMGateLayer:get_params() local pr = nerv.ParamRepo({self.bp}, self.loc_type) for i = 1, #self.dim_in do - pr:add(self["ltp" .. i].id, self["ltp" .. i]) + pr:add(self["ltp" .. i]) end return pr end diff --git a/nerv/layer/projection.lua b/nerv/layer/projection.lua index 9d376bd..077125b 100644 --- a/nerv/layer/projection.lua +++ b/nerv/layer/projection.lua @@ -64,7 +64,7 @@ end function ProjectionLayer:get_params() local pr = nerv.ParamRepo({self.ltp1}, self.loc_type) for i = 2, #self.dim_in do - pr:add(self["ltp" .. i].id, self["ltp" .. i]) + pr:add(self["ltp" .. i]) end return pr end |