Momentum_step — momentum_step • fastai

Step for SGD with momentum with `lr`

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

Arguments

p: p
lr: learning rate
grad_avg: grad average
...: additional arguments to pass

Value

None