torch.optim.lr_scheduler._LRScheduler Class Reference

Inheritance diagram for torch.optim.lr_scheduler._LRScheduler:

Public Member Functions
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
	optimizer
	base_lrs
	last_epoch

Detailed Description

Definition at line 10 of file lr_scheduler.py.

Member Function Documentation

def torch.optim.lr_scheduler._LRScheduler.load_state_dict	(		self,
			state_dict
	)

Loads the schedulers state.

Arguments:
    state_dict (dict): scheduler state. Should be an object returned
from a call to :meth:`state_dict`.

Definition at line 36 of file lr_scheduler.py.

def torch.optim.lr_scheduler._LRScheduler.state_dict

(

self

)

Returns the state of the scheduler as a :class:`dict`.

It contains an entry for every variable in self.__dict__ which
is not the optimizer.

Definition at line 28 of file lr_scheduler.py.

---

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

• torch/optim/lr_scheduler.py