zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-05 11:28:12 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix AdamW weight_decay default value in docstring (#2557)

Browse Source
This commit is contained in:
Artur Antonov
2025-09-01 07:29:30 +03:00
committed by GitHub
parent 8ce49cd39e
commit c5460762e7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/optimizers/optimizers.py
View File

@@ -556,7 +556,7 @@ class AdamW(Adam):
eps (float, optional): The term :math:`\epsilon` added to the eps (float, optional): The term :math:`\epsilon` added to the
denominator to improve numerical stability. Default: ``1e-8`` denominator to improve numerical stability. Default: ``1e-8``
weight_decay (float, optional): The weight decay :math:`\lambda`. weight_decay (float, optional): The weight decay :math:`\lambda`.
Default: ``0``. Default: ``0.01``.
bias_correction (bool, optional): If set to ``True``, bias correction bias_correction (bool, optional): If set to ``True``, bias correction
is applied. Default: ``False`` is applied. Default: ``False``
""" """