LRScheduler

class LRScheduler(optimizer, current_epoch=- 1)

Base class for all learning rate based schedulers.

Parameters

• optimizer (Optimizer) – wrapped optimizer.

• current_epoch (int) – the index of current epoch. Default: -1.

Returns

An instance of the LRScheduler.

get_lr()

Compute current learning rate for the scheduler.

load_state_dict(state_dict)

Loads the schedulers state.

Parameters

state_dict – scheduler state.

state_dict()

Returns the state of the scheduler as a dict. It contains an entry for every variable in self.__dict__ which is not the optimizer.

previous

LAMBFp16

next

MultiStepLR