1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
require 'torch'
optim = {}
-- optimizations
torch.include('optim', 'sgd.lua')
torch.include('optim', 'cg.lua')
torch.include('optim', 'asgd.lua')
torch.include('optim', 'nag.lua')
torch.include('optim', 'fista.lua')
torch.include('optim', 'lbfgs.lua')
torch.include('optim', 'adagrad.lua')
torch.include('optim', 'rprop.lua')
torch.include('optim', 'adam.lua')
torch.include('optim', 'adamax.lua')
torch.include('optim', 'rmsprop.lua')
torch.include('optim', 'adadelta.lua')
torch.include('optim', 'cmaes.lua')
-- line search functions
torch.include('optim', 'lswolfe.lua')
-- helpers
torch.include('optim', 'polyinterp.lua')
torch.include('optim', 'checkgrad.lua')
-- tools
torch.include('optim', 'ConfusionMatrix.lua')
torch.include('optim', 'Logger.lua')
return optim
|