Rate this Page

torch.optim.asgd.asgd#

torch.optim.asgd.asgd(params, grads, axs, mus, etas, state_steps, foreach=None, maximize=False, differentiable=False, capturable=False, has_complex=False, *, lambd, lr, t0, alpha, weight_decay)[source]#

Functional API that performs asgd algorithm computation.

See ASGD for details.