Correct? lbfgs eval

1 files changed, 6 insertions, 6 deletions

diff --git a/LBFGSOptimization.lua b/LBFGSOptimization.lua
index 30321eb..b8b028b 100644
--- a/LBFGSOptimization.lua
+++ b/LBFGSOptimization.lua
@@ -37,12 +37,12 @@ end
 
 function LBFGS:optimize()
    -- callback for lBFGS
-   lbfgs.evaluate = self.evaluate
-
-   -- allreduce sync
-   if self.allreduce then
-      allreduce.accumulate(self.parameters)
-   end
+   lbfgs.evaluate = function()
+                       self.evaluate()
+                       if self.allreduce then
+                          allreduce.accumulate(self.parameters)
+                       end
+                    end
 
    -- the magic function: will update the parameter vector according to the l-BFGS method
    self.output = lbfgs.run()