zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-30 22:51:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

format (#510)

Browse Source
This commit is contained in:
Awni Hannun 2024-01-20 10:33:46 -08:00 committed by GitHub
parent 363d3add6d
commit d52383367a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/mlx/optimizers.py
View File

@ -424,7 +424,9 @@ class Adamax(Adam):
):
super().__init__(learning_rate, betas, eps)
if not 0.0 <= eps:
raise ValueError(f"Epsilon value should be >=0, {self.eps} was provided instead")
raise ValueError(
f"Epsilon value should be >=0, {self.eps} was provided instead"
)
def apply_single(
self, gradient: mx.array, parameter: mx.array, state: OptimizerState