diff options
Diffstat (limited to 'nerv/layer/affine.lua')
-rw-r--r-- | nerv/layer/affine.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/nerv/layer/affine.lua b/nerv/layer/affine.lua index d56fcb8..566e9bc 100644 --- a/nerv/layer/affine.lua +++ b/nerv/layer/affine.lua @@ -127,4 +127,5 @@ function AffineLayer:get_params() for i = 2, #self.dim_in do pr:add(self["ltp" .. i].id, self["ltp" .. i]) end + return pr end |