utils.lr_scheduler.WarmupMultiFactorScheduler Class Reference

Inheritance diagram for utils.lr_scheduler.WarmupMultiFactorScheduler:

List of all members.

Public Member Functions
def	__call__
def	__init__
Public Attributes
	count
	cur_step_ind
	factor
	step
	warmup
	warmup_lr
	warmup_step

---

Detailed Description

Reduce learning rate in factor at steps specified in a list

Assume the weight has been updated by n times, then the learning rate will
be

base_lr * factor^(sum((step/n)<=1)) # step is an array

Parameters
----------
step: list of int
    schedule learning rate after n updates
factor: float
    the factor for reducing the learning rate

Definition at line 12 of file lr_scheduler.py.

---

Constructor & Destructor Documentation

def utils.lr_scheduler.WarmupMultiFactorScheduler.__init__	(		self,
			step,
			factor = 1,
			warmup = False,
			warmup_lr = 0,
			warmup_step = 0
	)

Definition at line 27 of file lr_scheduler.py.

---

Member Function Documentation

def utils.lr_scheduler.WarmupMultiFactorScheduler.__call__	(		self,
			num_update
	)

Call to schedule current learning rate

Parameters
----------
num_update: int
    the maximal number of updates applied to a weight.

Definition at line 45 of file lr_scheduler.py.

---

Member Data Documentation

utils::lr_scheduler.WarmupMultiFactorScheduler::count

Definition at line 27 of file lr_scheduler.py.

utils::lr_scheduler.WarmupMultiFactorScheduler::cur_step_ind

Definition at line 27 of file lr_scheduler.py.

utils::lr_scheduler.WarmupMultiFactorScheduler::factor

Definition at line 27 of file lr_scheduler.py.

utils::lr_scheduler.WarmupMultiFactorScheduler::step

Definition at line 27 of file lr_scheduler.py.

utils::lr_scheduler.WarmupMultiFactorScheduler::warmup

Definition at line 27 of file lr_scheduler.py.

utils::lr_scheduler.WarmupMultiFactorScheduler::warmup_lr

Definition at line 27 of file lr_scheduler.py.

utils::lr_scheduler.WarmupMultiFactorScheduler::warmup_step

Definition at line 27 of file lr_scheduler.py.

---

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

• lr_scheduler.py