average_grad {fastai}R 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


[Package fastai version 2.2.2 Index]