rms_prop_step {fastai}

Rms_prop_step

Description

Step for SGD with momentum with 'lr'

Usage

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