Skip to main content

burn_mamba/modules/loss/
mod.rs

1//! Loss functions used by the examples and tests.
2
3/// Binary cross-entropy.
4pub mod bce;
5/// Cross-entropy (multi-class).
6pub mod cross_entropy;
7/// Mean squared error.
8pub mod mse;