1 2 3 4 5 6 7
mod compute; mod decay; mod policy; mod warmup; pub use compute::{compute_learning_rate, compute_learning_rate_f64}; pub use policy::{LrSchedule, LrScheduleF64, ScheduleError};