This commit is contained in:
Awni Hannun
2024-01-20 10:33:46 -08:00
committed by GitHub
parent 363d3add6d
commit d52383367a

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