aboutsummaryrefslogtreecommitdiff
path: root/nerv/layer
diff options
context:
space:
mode:
Diffstat (limited to 'nerv/layer')
-rw-r--r--nerv/layer/affine.lua2
-rw-r--r--nerv/layer/lstm_gate.lua2
-rw-r--r--nerv/layer/projection.lua2
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