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

Add smoothed L1 loss and enhancements to cross entropy loss (#166)

Browse Source 
* Add smooth_l1_loss
* Add labels moothing for cross entropy loss

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
jojopuppet
2023-12-18 23:26:21 +08:00
committed by GitHub
parent 0e5807bbcb
commit 18cca64c81
3 changed files with 278 additions and 172 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,9 +9,10 @@ Loss Functions
:toctree: _autosummary_functions
:template: nn-module-template.rst
cross_entropy
binary_cross_entropy
cross_entropy
kl_div_loss
l1_loss
mse_loss
nll_loss
kl_div_loss
smooth_l1_loss