average_grad {fastai} | R Documentation |
Keeps track of the avg grads of 'p' in 'state' with 'mom'.
average_grad(p, mom, dampening = FALSE, grad_avg = NULL, ...)
p |
p |
mom |
momentum |
dampening |
dampening |
grad_avg |
grad average |
... |
additional args to pass |
None