dacapo.experiments.trainers.trainer_config
Module Contents
Classes
A class to represent the Trainer Configurations. |
- class dacapo.experiments.trainers.trainer_config.TrainerConfig
A class to represent the Trainer Configurations.
It is the base class for trainer configurations. Each subclass of a Trainer should have a specific config class derived from TrainerConfig.
- name
A unique name for this trainer.
- Type:
str
- batch_size
The batch size to be used during training.
- Type:
int
- learning_rate
The learning rate of the optimizer.
- Type:
float
- name: str
- batch_size: int
- learning_rate: float
- verify() Tuple[bool, str]
Verify whether this TrainerConfig is valid or not.
- Returns:
A tuple containing a boolean indicating whether the TrainerConfig is valid and a message explaining why.
- Return type:
tuple