From 05767b026fee6bd22c44f3745c9e0e13558968d7 Mon Sep 17 00:00:00 2001 From: Chunyang Wen Date: Thu, 28 Dec 2023 13:51:30 +0800 Subject: [PATCH] Add information for dropout probability (#304) Co-authored-by: Chunyang Wen --- python/mlx/nn/layers/dropout.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/mlx/nn/layers/dropout.py b/python/mlx/nn/layers/dropout.py index caa7a6452..8c7326e2d 100644 --- a/python/mlx/nn/layers/dropout.py +++ b/python/mlx/nn/layers/dropout.py @@ -19,7 +19,7 @@ class Dropout(Module): super().__init__() if p < 0 or p >= 1: - raise ValueError("The dropout probability should be in [0, 1)") + raise ValueError("The dropout probability {p} is not in [0, 1)") self._p_1 = 1 - p @@ -62,7 +62,7 @@ class Dropout2d(Module): super().__init__() if p < 0 or p >= 1: - raise ValueError("The dropout probability should be in [0, 1)") + raise ValueError(f"The dropout probability {p} is not in [0, 1)") self._p_1 = 1 - p