指数学习率 ¶
- 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。