momentum_step: Momentum_step

View source: R/optimizers.R

momentum_stepR Documentation

Momentum_step

Description

Step for SGD with momentum with 'lr'

Usage

momentum_step(p, lr, grad_avg, ...)

Arguments

p

p

lr

learning rate

grad_avg

grad average

...

additional arguments to pass

Value

None


fastai documentation built on March 31, 2023, 11:41 p.m.