| CallbackConfig |
Configuration for an optional shell script callback invoked after a checkpoint, export, or shutdown event |
| MetricsLogsConfig |
|
| ShutdownConfig |
Configuration for automatic training shutdown after a checkpoint, useful for preemptible jobs |
| StreamingTrainerCallbackConfig |
Trainer callback for online RL: exports and broadcasts model weights via Redis after each update |
| TrainerCallbackConfig |
Abstract base configuration for trainer callbacks that hook into training events |
| TrainerConfig (abstract) |
Abstract base configuration for a training run: model, data, schedule, optimizer, callbacks, and checkpointing |
| TrainingCheckpointBaseConfig (abstract) |
|
| TrainingCheckpointConfig |
Configuration for saving full training checkpoints (weights + optimizer state) at a fixed interval |
| TrainingConfig |
Configuration for training phases: iterations, checkpoints, exports, logging, evaluators, and W&B |
| TrainingExportConfig |
Configuration for exporting model weights to an external format (e.g. HuggingFace) at a fixed interval |
| WandbAlertConfig |
Configuration for periodic Weights & Biases status alerts during training |
| WandbConfig |
|
| WeightsBroadcastConfig |
|