Skip to content

main

main

Main Hydra configuration dataclass for the train endpoint.

MainConfig dataclass

MainConfig(defaults=(lambda: [{'experiment': ExperimentConfig}, {'task': 'base'}, {'training': 'base'}, {'policy': 'base'}, {'inference': 'base'}, {'optional quantization': None}])(), experiment=ExperimentConfig(), task=TaskSpaceConfig(), training=TrainingConfig(), policy=PolicyConfig(), quantization=None)

Root Hydra config composed from experiment, task, training, policy, and inference configs.

Attributes:

Name Type Description
defaults list[Any]

Hydra defaults list composed into the run config.

experiment ExperimentConfig

Experiment settings.

task TaskSpaceConfig

Task settings.

training TrainingConfig

Training settings.

policy PolicyConfig

Policy settings.

quantization Any

Quantization-aware training settings, or null.