MultiStepLR

class MultiStepLR(optimizer, milestones, gamma=0.1, current_epoch=- 1)

以gamma为倍率阶梯式衰减各参数组的学习率

当epoch的数目达到milestones之一时，才会执行。

参数

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

• milestones (Iter[int]) – epoch索引列表。必须按递增排序。

• gamma (float) – multiplicative factor of learning rate decay. Default: 0.1.

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

返回

An instance of the MultiStepLR Scheduler.

get_lr()

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

load_state_dict(state_dict)

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

参数

state_dict – 调度器状态

state_dict()

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

上一页

LRScheduler

下一页

megengine.optimizer.clip_grad_norm