megengine.optimizer.LRScheduler

class LRScheduler(optimizer, current_epoch=- 1)[源代码]

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

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

  • current_epoch (int) – 当前epoch的索引。默认:-1

方法

get_lr()

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

load_state_dict(state_dict)

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

state_dict()

dict 的形式返回调度器的状态。

step([epoch])