LRScheduler

class LRScheduler(optimizer: Optimizer, last_epoch=-1, verbose='deprecated')[source]

基础类:object

在优化过程中调整学习率。

方法总结

get_last_lr()

返回当前调度器最后计算的学习率。

get_lr()

使用可链接形式的调度器计算学习率。

load_state_dict(state_dict)

加载调度器的状态。

print_lr(is_verbose, group, lr[, epoch])

显示当前的学习率。

state_dict()

返回调度器的状态为一个dict

step([epoch])

执行一个步骤。

方法文档

Parameters:

优化器 (Optimizer)

get_last_lr() List[float][source]

返回当前调度器最后计算的学习率。

Return type:

List[float]

get_lr() List[float][source]

使用可链接形式的调度器计算学习率。

Return type:

List[float]

load_state_dict(state_dict: Dict[str, Any])[source]

加载调度器的状态。

Args:
state_dict (dict): scheduler state. Should be an object returned

来自对 state_dict() 的调用。

Parameters:

state_dict (Dict[str, Any])

print_lr(is_verbose: bool, group: Dict[str, Any], lr: float, epoch: int | None = None)[源代码]

显示当前的学习率。

自版本2.4起已弃用:print_lr() 已被弃用。请使用 get_last_lr() 来访问学习率。

Parameters:
state_dict()[source]

返回调度器的状态为一个dict

它包含 self.__dict__ 中每个变量的条目,这些变量不是优化器。

step(epoch: int | None = None)[source]

执行一个步骤。

Parameters:

epoch (int | None)