hodu_nn 0.1.6

hodu nn
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

//! 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::MAELoss, mse::MSELoss,
};
pub use crate::module::Module;
pub use crate::modules::{
    activation::*,
    conv::Conv2D,
    dropout::Dropout,
    linear::Linear,
    pooling::{AdaptiveAvgPool2d, AdaptiveMaxPool2d, AvgPool2d, MaxPool2d},
};
pub use crate::optimizer::Optimizer;
pub use crate::optimizers::adam::{Adam, AdamW};
pub use crate::{eval, train};