modularml.models.combinators

Classes

MergeModel(*args, **kwargs)

class modularml.models.combinators.MergeModel(*args: Any, **kwargs: Any)

Bases: BaseModel, Module

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_.