optimizer

package

v0.0.2 Latest Latest Go to latest Published: May 11, 2023 License: MIT Imports: 8 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

func NewAdagrad(lr, weightDecay, epsilon float64) *Adagrad

func (adagrad *Adagrad) Print()

func (adagrad *Adagrad) Save() *pb.Optimizer

func (opt Adagrad) SetLr(lr float64)

func (opt Adagrad) Update(grads, params []*params.Params)

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

func NewAdam(lr, weightDecay, beta1, beta2, epsilon float64) *Adam

func (adam *Adam) Print()

func (adam *Adam) Save() *pb.Optimizer

func (opt Adam) SetLr(lr float64)

func (opt Adam) Update(grads, params []*params.Params)

type Optimizer interface {
	Update(grads, params []*params.Params)
	Save() *pb.Optimizer
	Print()
	SetLr(lr float64)
}

func Load(opt *pb.Optimizer) Optimizer

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

func NewSGD(lr, weightDecay float64) *SGD

func (opt SGD) Print()

func (sgd *SGD) Save() *pb.Optimizer

func (opt SGD) SetLr(lr float64)

func (opt SGD) Update(grads, params []*params.Params)