From 63d2755759495db0941e2944827c8c5936e39a8f Mon Sep 17 00:00:00 2001 From: Qi Liu Date: Thu, 5 May 2016 21:08:07 +0800 Subject: fix param_repo:add bug --- nerv/layer/affine.lua | 2 +- nerv/layer/lstm_gate.lua | 2 +- 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 3bf5a11..467eac9 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 99bf3ca..3bb3cb9 100644 --- a/nerv/layer/lstm_gate.lua +++ b/nerv/layer/lstm_gate.lua @@ -89,7 +89,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 d99401c..c0b5638 100644 --- a/nerv/layer/projection.lua +++ b/nerv/layer/projection.lua @@ -58,7 +58,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 -- cgit v1.2.3