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

Added TeX line breaks to mlx.optimizers.Lion docstring (#595)

Browse Source 
Fixes the "misplaced &" MathJax error in documentation.
This commit is contained in:
nathan
2024-01-30 19:37:34 -08:00
committed by GitHub
parent 199aebcf77
commit bad67fec37
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/mlx/optimizers.py
View File

@@ -465,8 +465,8 @@ class Lion(Optimizer):
.. math:: .. math::
c_{t + 1} &= \beta_1 m_t + (1 - \beta_1) g_t c_{t + 1} &= \beta_1 m_t + (1 - \beta_1) g_t \\
m_{t + 1} &= \beta_2 m_t + (1 - \beta_2) g_t m_{t + 1} &= \beta_2 m_t + (1 - \beta_2) g_t \\
w_{t + 1} &= w_t - \eta (\text{sign}(c_t) + \lambda w_t) w_{t + 1} &= w_t - \eta (\text{sign}(c_t) + \lambda w_t)
Args: Args: