Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.optim.lr_scheduler._LRScheduler Class Reference
Inheritance diagram for torch.optim.lr_scheduler._LRScheduler:
torch.optim.lr_scheduler.CosineAnnealingLR torch.optim.lr_scheduler.ExponentialLR torch.optim.lr_scheduler.LambdaLR torch.optim.lr_scheduler.MultiStepLR torch.optim.lr_scheduler.StepLR test_optim.LegacyMultiStepLR test_optim.LegacyStepLR

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


Detailed Description

Definition at line 10 of file

Member Function Documentation

def torch.optim.lr_scheduler._LRScheduler.load_state_dict (   self,
Loads the schedulers state.

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

Definition at line 36 of file

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

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