exponential

Documentation

Index

• type Exponential
  • func New(init, final float64, iter int) *Exponential
  • func (d *Exponential) Decay(lr float64, t int) float64

Types

type Exponential

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

Exponential defines an exponential decay depending on the time step:

lr = exp((times - t) * log(lr) + log(final))

func New

func New(init, final float64, iter int) *Exponential

New returns a new Exponential decay optimizer.

func (*Exponential) Decay

func (d *Exponential) Decay(lr float64, t int) float64

Decay calculates the decay of the learning rate lr at time t.