paddlespeech.vector.training.scheduler 模块

class paddlespeech.vector.training.scheduler.CyclicLRScheduler(base_lr: float = 1e-08, max_lr: float = 0.001, step_size: int = 10000)[来源]

基础: LRScheduler

方法

__call__()

返回当前周期最新计算的学习率。

get_lr()

对于那些重载LRScheduler(基类)的子类,用户应该有get_lr()的自定义实现。

set_dict(state_dict)

加载调度器的状态。

set_state_dict(state_dict)

加载调度器的状态。

state_dict()

返回调度程序的状态,格式为 dict

state_keys()

对于那些重载 LRScheduler(基类)的子类。

step()

step 应在 optimizer.step 之后调用。

get_lr()[来源]

对于那些重载 LRScheduler(基类)的子类,用户应该有一个自定义的 get_lr() 实现。

否则,将抛出一个 NotImplementedError 异常。

step()[来源]

step 应该在 optimizer.step 之后调用。它将根据当前的 epoch 更新优化器中的学习率。 新的学习率将在下一个 optimizer.step 中生效。

Args:

epoch (int, None):指定当前的轮次。默认值:无。从 last_epoch=-1 开始自动递增。

Returns: