training

training

Training and optimizer config registrations.

register

register(cs)

Store this domain's config nodes.

Parameters:

Name	Type	Description	Default
cs	ConfigStore	The global Hydra config store.	required

Source code in src/versatil/configs/store/training.py

def register(cs: ConfigStore) -> None:
    """Store this domain's config nodes.

    Args:
        cs: The global Hydra config store.
    """
    cs.store(group="training", name="base", node=TrainingConfig)
    cs.store(group="training/optimizer", name="base", node=OptimizerConfig)
    cs.store(group="training/optimizer", name="adamw_schema", node=AdamWConfig)
    cs.store(group="training/optimizer", name="adam_schema", node=AdamConfig)
    cs.store(group="training/optimizer", name="sgd_schema", node=SGDConfig)
    cs.store(
        group="training/optimizer/parameter_group",
        name="base",
        node=ParameterGroupConfig,
    )