diff options
Diffstat (limited to 'test/test_lbfgs.lua')
| -rw-r--r-- | test/test_lbfgs.lua | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/test/test_lbfgs.lua b/test/test_lbfgs.lua index 88e5b9a..f882361 100644 --- a/test/test_lbfgs.lua +++ b/test/test_lbfgs.lua @@ -1,4 +1,5 @@ dofile('rosenbrock.lua') +dofile('l2.lua') require 'liblbfgs' neval = 0 @@ -8,10 +9,12 @@ linesearch = 2 sparsity = 0 verbose = 2 nparam = 8 +local testfunc = rosenbrock local parameters = torch.Tensor(nparam):fill(0.1) +local gradParameters = torch.Tensor(nparam):zero() -output, gradParameters = rosenbrock(parameters) +output, gradParameters = testfunc(parameters,gradParameters) function printstats () print('nEval: '..neval) @@ -32,7 +35,7 @@ print('Starting:') printstats() lbfgs.evaluate = function() - output, gradParameters = rosenbrock(parameters) + output, gradParameters = testfunc(parameters,gradParameters) neval = neval + 1 printstats() return output |