xnn.common.config.schema.Config#

class xnn.common.config.schema.Config(model=<factory>, data=<factory>, optim=<factory>, device='auto', seed=1234, output_dir='runs/exp')[source]#

Bases: object

Top-level experiment configuration.

This nested dataclass tree is the single internal representation that every frontend loader (YAML, argparse, Hydra) produces; see loaders.py.

Variables:

  • model (ModelConfig) – Model architecture and hyperparameters. Defaults to a fresh ModelConfig.

  • data (DataConfig) – Dataset paths and data-loading options. Defaults to a fresh DataConfig.

  • optim (OptimConfig) – Optimizer and loss settings. Defaults to a fresh OptimConfig.

  • device (str) – Compute device (auto / cpu / cuda / cuda:0 …). Defaults to "auto".

  • seed (int) – Random seed for reproducibility. MACE seed. Defaults to 1234.

  • output_dir (str) – Directory for run outputs (checkpoints, logs). Defaults to "runs/exp".

Parameters:
On this page
Show Source