pub fn create_step_lr_trainer(
network: LSTMNetwork,
learning_rate: f64,
step_size: usize,
gamma: f64,
) -> ScheduledLSTMTrainer<MSELoss, SGD, StepLR>Expand description
Create a scheduled trainer with SGD and StepLR scheduler
pub fn create_step_lr_trainer(
network: LSTMNetwork,
learning_rate: f64,
step_size: usize,
gamma: f64,
) -> ScheduledLSTMTrainer<MSELoss, SGD, StepLR>Create a scheduled trainer with SGD and StepLR scheduler