zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-14 20:41:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add information for dropout probability (#304)

Browse Source 
Co-authored-by: Chunyang Wen <chunyang_wen@apple.com>
This commit is contained in:
Chunyang Wen 2023-12-28 13:51:30 +08:00 committed by GitHub
parent a83d5d60bd
commit 05767b026f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/mlx/nn/layers/dropout.py
View File

@ -19,7 +19,7 @@ class Dropout(Module):
super().__init__() super().__init__()
if p < 0 or p >= 1: 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 self._p_1 = 1 - p
@ -62,7 +62,7 @@ class Dropout2d(Module):
super().__init__() super().__init__()
if p < 0 or p >= 1: 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 self._p_1 = 1 - p