Rms_prop_step — rms_prop_step • fastai

Step for SGD with momentum with `lr`

rms_prop_step(p, lr, sqr_avg, eps, grad_avg = NULL, ...)

Arguments

p: p
lr: learning rate
sqr_avg: sqr average
eps: epsilon
grad_avg: grad average
...: additional arguments to pass

Value

None