diff options
| -rw-r--r-- | checkgrad.lua | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/checkgrad.lua b/checkgrad.lua index aecb969..402d9fc 100644 --- a/checkgrad.lua +++ b/checkgrad.lua @@ -27,11 +27,12 @@ function optim.checkgrad(opfunc, x, eps) local eps = eps or 1e-7 local dC_est = torch.Tensor():typeAs(dC):resizeAs(dC) for i = 1,dC:size(1) do + local tmp = x[i] x[i] = x[i] + eps local C1 = opfunc(x) x[i] = x[i] - 2 * eps local C2 = opfunc(x) - x[i] = x[i] + eps + x[i] = tmp dC_est[i] = (C1 - C2) / (2 * eps) end |