torch.optim.lr_scheduler.ExponentialLR Class Reference

Inheritance diagram for torch.optim.lr_scheduler.ExponentialLR:

Public Member Functions
def	__init__ (self, optimizer, gamma, last_epoch=-1)
def	get_lr (self)
Public Member Functions inherited from torch.optim.lr_scheduler._LRScheduler
def	__init__ (self, optimizer, last_epoch=-1)
def	state_dict (self)
def	load_state_dict (self, state_dict)
def	get_lr (self)
def	step (self, epoch=None)

Public Attributes
	gamma
	last_epoch
Public Attributes inherited from torch.optim.lr_scheduler._LRScheduler
	optimizer
	base_lrs
	last_epoch

Detailed Description

Decays the learning rate of each parameter group by gamma every epoch.
When last_epoch=-1, sets initial lr as lr.

Args:
    optimizer (Optimizer): Wrapped optimizer.
    gamma (float): Multiplicative factor of learning rate decay.
    last_epoch (int): The index of last epoch. Default: -1.

Definition at line 201 of file lr_scheduler.py.

---

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

• torch/optim/lr_scheduler.py