mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-25 15:28:08 +08:00

According the definition of smooth_l1_loss, the line diff = predictions - targets Should be updated to diff = mx.abs(predictions - targets) After the modification, the result is consistent with PyTorch smooth_l1_loss