kempnerforge.config.job¶

Top-level job configuration aggregating all sub-configs.

Classes

Top-level configuration aggregating all sub-configs.

class kempnerforge.config.job.JobConfig[source]¶

Bases: object

Top-level configuration aggregating all sub-configs.

model: ModelConfig¶

train: TrainConfig¶

optimizer: OptimizerConfig¶

scheduler: SchedulerConfig¶

data: DataConfig¶

eval: EvalConfig¶

distributed: DistributedConfig¶

checkpoint: CheckpointConfig¶

metrics: MetricsConfig¶

profiling: ProfilingConfig¶

validate(world_size=1)[source]¶

Run cross-config validations.

Parameters:: world_size (int)
Return type:: None

__init__(model=<factory>, train=<factory>, optimizer=<factory>, scheduler=<factory>, data=<factory>, eval=<factory>, distributed=<factory>, checkpoint=<factory>, metrics=<factory>, profiling=<factory>)¶

Parameters:

model (ModelConfig)
train (TrainConfig)
optimizer (OptimizerConfig)
scheduler (SchedulerConfig)
data (DataConfig)
eval (EvalConfig)
distributed (DistributedConfig)
checkpoint (CheckpointConfig)
metrics (MetricsConfig)
profiling (ProfilingConfig)

Return type:

None