utils.schemas.training

utils.schemas.training

Pydantic models for training hyperparameters

Classes

Name	Description
HyperparametersConfig	Training hyperparams configuration subset
JaggedLRConfig	JaggedLR configuration subset, can be used w/ ReLoRA training
LrGroup	Custom learning rate group configuration

HyperparametersConfig

utils.schemas.training.HyperparametersConfig()

Training hyperparams configuration subset

JaggedLRConfig

utils.schemas.training.JaggedLRConfig()

JaggedLR configuration subset, can be used w/ ReLoRA training

LrGroup

utils.schemas.training.LrGroup()

Custom learning rate group configuration