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

Public Member Functions

def __init__ (self, optimizer, T_max, eta_min=0, 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

 T_max
 
 eta_min
 
 last_epoch
 
- Public Attributes inherited from torch.optim.lr_scheduler._LRScheduler
 optimizer
 
 base_lrs
 
 last_epoch
 

Detailed Description

Definition at line 222 of file lr_scheduler.py.


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