Merge pull request #343 from GeorgOstrovski/master

Replace Module.flatten by nn.Module.flatten

1 files changed, 2 insertions, 2 deletions

diff --git a/Module.lua b/Module.lua
index f5029e9..f65e1b1 100644
--- a/Module.lua
+++ b/Module.lua
@@ -219,9 +219,9 @@ function Module:getParameters()
    -- get parameters
    local parameters,gradParameters = self:parameters()
    -- flatten parameters and gradients
-   local flatParameters = Module.flatten(parameters)
+   local flatParameters = nn.Module.flatten(parameters)
    collectgarbage()
-   local flatGradParameters = Module.flatten(gradParameters)
+   local flatGradParameters = nn.Module.flatten(gradParameters)
    collectgarbage()
 
    -- return new flat vector that contains all discrete parameters