Expand description
Train module
Modules§
- The checkpoint module.
- The logger module.
- The metric module.
- Renderer modules to display metrics and training information.
- The trainer module.
Structs§
- Simple classification output adapted for multiple metrics.
- This struct is used to install a local file application logger to output logs to a given file path.
- Learner struct encapsulating all components necessary to train a Neural Network model.
- Struct to configure and create a learner.
- Detailed training summary.
- An early stopping strategy based on a metrics collected during training or validation.
- Contains the metric value at a given time.
- Contains the summary of recorded values for a given metric.
- Multi devices train step.
- Multi-label classification output adapted for multiple metrics.
- Simple regression output adapted for multiple metrics.
- Contains the summary of recorded metrics for the training and validation steps.
- A training epoch.
- A training output.
- A handle that allows aborting the training process early.
- A validation epoch.
Enums§
- The condition that early stopping strategies should follow.
Traits§
- This trait is used to install an application logger.
- A strategy that checks if the training should be stopped.
- Trait to be implemented for training models.
- Trait to be implemented for validating models.