Rate this Page
★★★★★

torch.optim.adamax.adamax#

torch.optim.adamax.adamax(params,grads,exp_avgs,exp_infs,state_steps,foreach=None,maximize=False,differentiable=False,capturable=False,has_complex=False,*,eps,beta1,beta2,lr,weight_decay)[source]#

Functional API that performs adamax algorithm computation.

SeeAdamax for details.