LRScheduler

class LRScheduler(optimizer, current_epoch=- 1)

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

参数

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

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

返回

An instance of the LRScheduler.

get_lr()

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

load_state_dict(state_dict)

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

参数

state_dict – 调度器状态

state_dict()

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

上一页

LAMBFp16

下一页

MultiStepLR