utils.schemas.training

utils.schemas.training

Pydantic models for training hyperparameters

Classes

Name Description
HyperparametersConfig Training hyperparams configuration subset
LrGroup Custom learning rate group configuration

HyperparametersConfig

utils.schemas.training.HyperparametersConfig()

Training hyperparams configuration subset

LrGroup

utils.schemas.training.LrGroup()

Custom learning rate group configuration