nesterov

Documentation

Index

• type Nesterov
  • func MustNew(eta, rho float64) *Nesterov
  • func New(eta, rho float64) (*Nesterov, error)
  • func (o *Nesterov) Extend(n int)
  • func (o *Nesterov) Update(out, p *mat.VecDense, g mat.Vector) *mat.VecDense

Types

type Nesterov

type Nesterov struct {
	// contains filtered or unexported fields
}

Nesterov accelerated gradient descent optimiser; see the following link for details: https://fluxml.ai/Flux.jl/stable/training/optimisers/

func MustNew

func MustNew(eta, rho float64) *Nesterov

func New

func New(eta, rho float64) (*Nesterov, error)

func (*Nesterov) Extend

func (o *Nesterov) Extend(n int)

func (*Nesterov) Update

func (o *Nesterov) Update(out, p *mat.VecDense, g mat.Vector) *mat.VecDense