diff options
-rw-r--r-- | Module.lua | 3 | ||||
-rw-r--r-- | Sequential.lua | 6 |
2 files changed, 9 insertions, 0 deletions
@@ -131,6 +131,9 @@ function Module:cuda() return self:type('torch.CudaTensor') end +function Module:reset() +end + function Module:getParameters() -- get parameters local parameters,gradParameters = self:parameters() diff --git a/Sequential.lua b/Sequential.lua index 3e23350..a368964 100644 --- a/Sequential.lua +++ b/Sequential.lua @@ -89,6 +89,12 @@ function Sequential:share(mlp,...) end end +function Sequential:reset(stdv) + for i=1,#self.modules do + self.modules[i]:reset(stdv) + end +end + function Sequential:parameters() local function tinsert(to, from) if type(from) == 'table' then |