cli.logic.train package
cli.logic.train.train module
cli.logic.train.trainer module
- class cli.logic.train.trainer.CLITrainer(config: TrainingCliConfigType)
Bases:
object
Trainer class for the CLI. This module parses the CLI arguments and trains the model.
- Parameters:
config (TrainingCliConfigType) – The configuration for training.
- config
The configuration for training.
- Type:
- train()
The main training function. This function parses the CLI arguments and executes the training.