average_grad: Average_grad

View source: R/optimizers.R

average_gradR Documentation

Average_grad

Description

Keeps track of the avg grads of 'p' in 'state' with 'mom'.

Usage

average_grad(p, mom, dampening = FALSE, grad_avg = NULL, ...)

Arguments

p

p

mom

momentum

dampening

dampening

grad_avg

grad average

...

additional args to pass

Value

None


fastai documentation built on June 22, 2024, 11:15 a.m.