Optimizers

Optimizers

Optimizers provides a way to update the weights of Merlin.Var.

x = zerograd(rand(Float32,5,4))
opt = SGD(0.001)
opt(x)
println(x.grad)
Merlin.AdaGrad — Type.
AdaGrad

AdaGrad Optimizer.

References

source
Merlin.Adam — Type.
Adam

Adam Optimizer

References

source
Merlin.SGD — Type.
SGD

Stochastic Gradient Descent Optimizer.

Arguments

  • rate: learning rate

  • [momentum=0.0]: momentum coefficient

  • [nesterov=false]: use nesterov acceleration or not

source