yscv_optim/
core.rs

1//! Optimizers and training helpers for yscv models.
2#![deny(unsafe_code)]
3
4pub const CRATE_ID: &str = "yscv-optim";
5
6#[path = "adagrad.rs"]
7mod adagrad;
8#[path = "adam.rs"]
9mod adam;
10#[path = "adamw.rs"]
11mod adamw;
12#[path = "clip.rs"]
13mod clip;
14#[path = "error.rs"]
15mod error;
16#[path = "lamb.rs"]
17mod lamb;
18#[path = "lars.rs"]
19mod lars;
20#[path = "lookahead.rs"]
21mod lookahead;
22#[path = "lr.rs"]
23mod lr;
24#[path = "radam.rs"]
25mod radam;
26#[path = "rmsprop.rs"]
27mod rmsprop;
28#[path = "scheduler.rs"]
29mod scheduler;
30#[path = "sgd.rs"]
31mod sgd;
32#[path = "validate.rs"]
33mod validate;
34
35pub use adagrad::Adagrad;
36pub use adam::Adam;
37pub use adamw::AdamW;
38pub use clip::{clip_grad_norm_, clip_grad_value_};
39pub use error::OptimError;
40pub use lamb::Lamb;
41pub use lars::Lars;
42pub use lookahead::{Lookahead, StepOptimizer};
43pub use lr::LearningRate;
44pub use radam::RAdam;
45pub use rmsprop::RmsProp;
46pub use scheduler::{
47    CosineAnnealingLr, CosineAnnealingWarmRestarts, CyclicLr, ExponentialLr, LambdaLr,
48    LinearWarmupLr, LrScheduler, MultiStepLr, OneCycleLr, PolynomialDecayLr, ReduceLrOnPlateau,
49    StepLr,
50};
51pub use sgd::Sgd;
52
53#[path = "tests.rs"]
54#[cfg(test)]
55mod tests;
yscv_optim/core.rs

yscv_optim/
core.rs
