zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-18 10:26:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixes (#131)

Browse Source
This commit is contained in:
Awni Hannun
2023-12-11 09:26:49 -08:00
committed by GitHub
parent f4ddd7dc44
commit 69505b4e9b
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/nn/losses.py
View File

@@ -203,4 +203,4 @@ def _reduce(loss: mx.array, reduction: str = "none"):
elif reduction == "none":
return loss
else:
raise ValueError("Invalid reduction. Must be 'none', 'mean', or 'sum'.")
raise ValueError("Invalid reduction. Must be 'none', 'mean', or 'sum'.")