1 files changed, 2 insertions, 0 deletions
diff --git a/Module.lua b/Module.lua
index bc5a7fb..ab06636 100644
--- a/Module.lua
+++ b/Module.lua
@@ -241,7 +241,9 @@ function Module:getParameters()
 
    -- flatten parameters and gradients
    local flatParameters = flatten(parameters)
+   collectgarbage()
    local flatGradParameters = flatten(gradParameters)
+   collectgarbage()
 
    -- return new flat vector that contains all discrete parameters
    return flatParameters, flatGradParameters