Corrected the example of value_and_grad (#274)

* Corrected the example for mx.value_and_grad * Reformat through pre-commit/black
2025-09-18 10:26:56 +08:00 · 2023-12-24 00:36:38 +05:30
parent f91f450141
commit acf1721b98
2 changed files with 2 additions and 4 deletions
--- a/python/mlx/nn/layers/dropout.py
+++ b/python/mlx/nn/layers/dropout.py
@@ -88,5 +88,3 @@ class Dropout2d(Module):

        mask = mx.random.bernoulli(p=self._p_1, shape=mask_shape)
        return (1 / self._p_1) * mask * x
-
-