zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add Hinge, Huber and LogCosh losses (#199)

Browse Source
This commit is contained in:
Nicholas Santavas
2023-12-22 19:28:10 +01:00
committed by GitHub
parent e8deca84e0
commit d35fa1db41
4 changed files with 117 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docs/src/python/nn/losses.rst
View File

@@ -16,4 +16,7 @@ Loss Functions
mse_loss
nll_loss
smooth_l1_loss
triplet_loss
triplet_loss
hinge_loss
huber_loss
log_cosh_loss