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]