Fixed eval counter for SGFixed eval counter for SGDD

1 files changed, 3 insertions, 1 deletions

diff --git a/sgd.lua b/sgd.lua
index d56b8d1..280f8fb 100644
--- a/sgd.lua
+++ b/sgd.lua
@@ -21,7 +21,6 @@ function optim.sgd(opfunc, x, state)
    local mom = state.momentum or 0
    local lrs = state.learningRates
    state.evalCounter = state.evalCounter or 0
-   state.evalCounter = state.evalCounter + 1
    local nevals = state.evalCounters
 
    -- (1) evaluate f(x) and df/dx
@@ -56,6 +55,9 @@ function optim.sgd(opfunc, x, state)
       x:add(-clr, state.dfdx)
    end
 
+   -- (5) update evaluation counter
+   state.evalCounter = state.evalCounter + 1
+
    -- return f(x_old), and x_new
    return x,fx
 end