optim/init.lua at master · YitzhakSp/optim · GitHub

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