lr_scheduler.rst 1.0 KB

12345678910111213141516171819202122232425262728293031323334
  1. .. role:: hidden
  2. :class: hidden-section
  3. .. _Learning Rate Schedulers:
  4. Learning Rate Schedulers
  5. ========================
  6. Learning Rate Schedulers update the learning rate over the course of training.
  7. Learning rates can be updated after each update via :func:`step_update` or at
  8. epoch boundaries via :func:`step`.
  9. .. automodule:: fairseq.optim.lr_scheduler
  10. :members:
  11. .. autoclass:: fairseq.optim.lr_scheduler.FairseqLRScheduler
  12. :members:
  13. :undoc-members:
  14. .. autoclass:: fairseq.optim.lr_scheduler.cosine_lr_scheduler.CosineSchedule
  15. :members:
  16. :undoc-members:
  17. .. autoclass:: fairseq.optim.lr_scheduler.fixed_schedule.FixedSchedule
  18. :members:
  19. :undoc-members:
  20. .. autoclass:: fairseq.optim.lr_scheduler.inverse_square_root_schedule.InverseSquareRootSchedule
  21. :members:
  22. :undoc-members:
  23. .. autoclass:: fairseq.optim.lr_scheduler.reduce_lr_on_plateau.ReduceLROnPlateau
  24. :members:
  25. :undoc-members:
  26. .. autoclass:: fairseq.optim.lr_scheduler.triangular_lr_scheduler.TriangularSchedule
  27. :members:
  28. :undoc-members: