hodu_nn 0.1.5

hodu nn
Documentation 
1
2
3
4
5
6
7
8
9
10

//! Prelude module for convenient imports
//!
//! Usage: `use hodu_nn::prelude::*;`

// Re-export core library types and functions
pub use crate::losses::{binary_cross_entropy::BCEWithLogitsLoss, cross_entropy::CrossEntropyLoss, mae::MAE, mse::MSE};
pub use crate::module::Module;
pub use crate::modules::{activation::*, conv::Conv2D, linear::Linear};
pub use crate::optimizer::Optimizer;
pub use crate::optimizers::adam::{Adam, AdamW};