diff options
author | Ke Tran <ketranm@users.noreply.github.com> | 2015-03-29 19:07:22 +0300 |
---|---|---|
committer | Ke Tran <ketranm@users.noreply.github.com> | 2015-03-29 19:07:22 +0300 |
commit | 8621290e56bc7341dba9442921017f4d129cfa29 (patch) | |
tree | 04ec91cd0ce7af8c8d570b815b67bf66842049f3 | |
parent | 58e2b595fccf326881f67fa4777c549cb5359fbc (diff) |
test Adadelta implementation
-rw-r--r-- | test/test_adadelta.lua | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/test/test_adadelta.lua b/test/test_adadelta.lua new file mode 100644 index 0000000..9bf5144 --- /dev/null +++ b/test/test_adadelta.lua @@ -0,0 +1,23 @@ +require 'torch' +require 'optim' +require 'adadelta' +require 'rosenbrock' +require 'l2' + +x = torch.Tensor(2):fill(0) +fx = {} +state = {} +config = {p=0.9, eps=1e-8, learningRate=1e-3} +for i = 1,10001 do + x,f=optim.adadelta(rosenbrock,x,config,state) + if (i-1)%1000 == 0 then + table.insert(fx,f[1]) + end +end + +print() +print('Rosenbrock test') +print() +print('x=');print(x) +print('fx=') +for i=1,#fx do print((i-1)*1000+1,fx[i]); end |