R/optimizers.R
average_grad.Rd
Keeps track of the avg grads of `p` in `state` with `mom`.
average_grad(p, mom, dampening = FALSE, grad_avg = NULL, ...)
p
momentum
dampening
grad average
additional args to pass
None