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

* Add smooth_l1_loss * Add labels moothing for cross entropy loss --------- Co-authored-by: Awni Hannun <[email protected]>
2023-12-18 07:26:21 -08:00
parent 0e5807bbcb
commit 18cca64c81
3 changed files with 278 additions and 172 deletions
@@ -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