attribute
logger
Pythonlogger = logging.getLogger(__name__)This module provides utility functions to compose a sequential set of networks/layers.
Parameters are in helia_edge.models.composer_params.
Functions
| Name | Description |
|---|---|
build |
Composer model from ComposerParams |
composer_layer |
Composes a sequential set of networks/layers |
loggerattributecomposer_layerfunctionComposes a sequential set of networks/layers.buildfunctionBuild a Composer model.logger = logging.getLogger(__name__)Composes a sequential set of networks/layers.
composer_layer(x: keras.KerasTensor, params: ComposerParams) -> keras.KerasTensorComposes a sequential set of networks/layers. Useful for adding custom layers to a pre-trained model (e.g. foundation).
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
x | keras.KerasTensor | Required | Model input |
params | ComposerParams | Required | Model parameters |
Returns
| Type | Description |
|---|---|
keras.KerasTensor | keras.KerasTensor: Model output |
Build a Composer model.
build( params: ComposerParams, input_shape: tuple[int | None, ...], *, batch_size: int | None = None, name: str | None = None,) -> keras.ModelBuild a Composer model.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
params | ComposerParams | Required | Model parameters. |
input_shape | tuple[int | None, ...] | Required | Input shape without the batch axis; None for a variable axis. |
batch_size | int | None | None | Static batch size; None for a dynamic batch. |
name | str | None | None | Model name; the family when None. |
Returns
| Type | Description |
|---|---|
keras.Model | keras.Model: The model, named ``composer`` unless ``name`` is given. |