快捷键

指数学习率 ¶

class torch.optim.lr_scheduler.ExponentialLR(optimizer, gamma, last_epoch=- 1)[source][source]

每个参数组在每个 epoch 通过 gamma 衰减学习率。

当 last_epoch=-1 时,设置初始 lr 为 lr。

参数:
  • 优化器(Optimizer)- 包装的优化器。

  • gamma(浮点数)- 学习率衰减的乘法因子。

  • last_epoch(整数)- 上一个 epoch 的索引。默认:-1。

get_last_lr()[source]

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

返回类型:

list[float]

get_lr()[source][source]

计算每个参数组的学习率。

load_state_dict(state_dict)[source]

加载调度器的状态。

参数:

state_dict (dict) – 调度器状态。应为调用 state_dict() 返回的对象。

state_dict()[source]

返回调度器的状态作为 dict

包含了 self.__dict__中每个非优化器的变量条目。

step(epoch=None)[source]

执行一步操作。


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源