← Back to API reference

neuraltrain.models.base.BaseModelConfig

pydantic model neuraltrain.models.base.BaseModelConfig

Base class for model configurations.

Two families share this base:

• Brain-model configs – the top-level models built from a build context. They subclass BaseBrainModelConfig and implement build(...) with context-named parameters; the factory-facing build_from_context(ctx) is provided for free.

• Sub-component configs (e.g. transformers, mergers, MLP heads) which are composed manually inside larger models and whose build takes bespoke arguments.

Both families therefore implement build; it is kept permissive on this base so subclasses can override it with their own signatures.

build(*args: Any, **kwargs: Any) → Module

← Back to API reference