lmflow.optim.nadam#

Classes#

Module Contents#

class lmflow.optim.nadam.NAdam(params, lr=0.002, betas=(0.9, 0.999), eps=1e-08, weight_decay=0, momentum_decay=0.004)[source]#

Bases: torch.optim.Optimizer

__setstate__(state)[source]#
step(closure=None)[source]#