LRScheduler

class LRScheduler(optimizer, current_epoch=- 1)

所有学习率调度器的基类。

参数:

• optimizer (Optimizer) – 包装后的优化器。

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

get_lr()

计算当前调度器(scheduler)的学习率。

load_state_dict(state_dict)

加载调度器(scheduler)的状态。

参数:

state_dict – 调度器状态

state_dict()

以 dict 返回调度器状态。 包含 self.__dict__ 中的所有变量，注意调度器这不是优化器。