pytorch-lightning
afa7d56e - Add learning rate scheduling support for `DeepSpeedStrategy` (#20320)

Commit
211 days ago
Add learning rate scheduling support for `DeepSpeedStrategy` (#20320) * Update fabric.py * Update deepspeed.py * Update fsdp.py * Update strategy.py * Update xla_fsdp.py --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Luca Antiga <luca.antiga@gmail.com> Co-authored-by: Luca Antiga <luca@lightning.ai> Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> Co-authored-by: Jirka B <j.borovec+github@gmail.com>
Author
Parents
Loading