mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-01 04:24:36 +08:00
Add cosine_annealing_warm_restarts to scheduler documentation
This commit is contained in:
@@ -8,6 +8,7 @@ Schedulers
|
||||
.. autosummary::
|
||||
:toctree: _autosummary
|
||||
|
||||
cosine_annealing_warm_restarts
|
||||
cosine_decay
|
||||
exponential_decay
|
||||
join_schedules
|
||||
|
Reference in New Issue
Block a user