mirror of
https://github.com/ml-explore/mlx.git
synced 2025-12-16 01:49:05 +08:00
3b661b7394
* Add linear warmup to schedules for use with existing schedules * Changed parameters for simplicity of most common case (0 initial value) * Added ScheduleJoiner and updated documentation * ScheduleJoiner -> join_schedules (ala optax #) * black compliance * Different evaluation of schedules * nits --------- Co-authored-by: Awni Hannun <awni@apple.com>
16 lines
219 B
ReStructuredText
16 lines
219 B
ReStructuredText
.. _schedulers:
|
|
|
|
Schedulers
|
|
==========
|
|
|
|
.. currentmodule:: mlx.optimizers
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
cosine_decay
|
|
exponential_decay
|
|
join_schedules
|
|
linear_schedule
|
|
step_decay
|