modularml.models.base

Classes

BaseModel(config, backend)

class modularml.models.base.BaseModel(config: Dict[str, Any], backend: Backend)

Bases: ABC

abstractmethod build(input_shape: Tuple[int] | None = None, output_shape: Tuple[int] | None = None) → None

Build the internal model layers given an input shape.

abstractmethod forward(*args, **kwargs)

Forward pass.

classmethod from_config(config: Dict[str, Any]) → BaseModel

Dynamically reconstructs a model from config. Prioritizes _target_ path, falls back to ModelRegistry.

get_config() → Dict[str, Any]

Return a serializable config dictionary, including _target_.