create_one_cycle_trainer

rust_lstm::training

Function create_one_cycle_trainer 

Source 
pub fn create_one_cycle_trainer(
    network: LSTMNetwork,
    max_lr: f64,
    total_steps: usize,
) -> ScheduledLSTMTrainer<MSELoss, Adam, OneCycleLR>
Expand description

Create a scheduled trainer with Adam and OneCycleLR scheduler