Crate ghostflow_optim

Crate ghostflow_optim 

Source
Expand description

GhostFlow Optimizers

Optimization algorithms for training neural networks.

Re-exports§

pub use optimizer::Optimizer;
pub use sgd::SGD;
pub use adam::Adam;
pub use adam::AdamW;
pub use scheduler::*;

Modules§

adam
Adam and AdamW optimizers
optimizer
Base Optimizer trait
scheduler
Learning rate schedulers
sgd
Stochastic Gradient Descent optimizer