diff --git a/python/mlx/nn/layers/dropout.py b/python/mlx/nn/layers/dropout.py index e2cc981e2..14c5cb15e 100644 --- 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 - -