torch.optim.sparse_adam.SparseAdam Class Reference

Inheritance diagram for torch.optim.sparse_adam.SparseAdam:

Public Member Functions
def	__init__ (self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8)
def	step (self, closure=None)

Detailed Description

Definition at line 6 of file sparse_adam.py.

Member Function Documentation

def torch.optim.sparse_adam.SparseAdam.step	(		self,
			closure = None
	)

Performs a single optimization step.

Arguments:
    closure (callable, optional): A closure that reevaluates the model
and returns the loss.

Definition at line 37 of file sparse_adam.py.

---

The documentation for this class was generated from the following file:

• torch/optim/sparse_adam.py