Module training 

Structs

EarlyStopper

Early stopping state tracker

EarlyStoppingConfig

Configuration for early stopping

LSTMBatchTrainer

Batch trainer for LSTM networks with configurable loss and optimizer Processes multiple sequences simultaneously for improved performance

LSTMTrainer

Main trainer for LSTM networks with configurable loss and optimizer

ScheduledLSTMTrainer

Specialized trainer for scheduled optimizers that automatically steps the scheduler

TrainingConfig

Configuration for training hyperparameters

TrainingMetrics

Training metrics tracked during training

Enums

EarlyStoppingMetric

Metric to monitor for early stopping

Functions

create_adam_batch_trainer

Create a batch trainer with Adam optimizer and MSE loss

create_basic_batch_trainer

Create a basic batch trainer with SGD optimizer and MSE loss

create_basic_trainer

Create a basic trainer with SGD optimizer and MSE loss

create_cosine_annealing_trainer

Create a scheduled trainer with Adam and CosineAnnealingLR scheduler

create_one_cycle_trainer

Create a scheduled trainer with Adam and OneCycleLR scheduler

create_step_lr_trainer

Create a scheduled trainer with SGD and StepLR scheduler